setup install

This commit is contained in:
Tryndaron
2024-04-02 13:56:13 +02:00
parent 5ee519d466
commit d458c28a0f
3 changed files with 3 additions and 122 deletions
+2 -2
View File
@@ -1,7 +1,7 @@
import pytest
import os
from unittest import mock
from scraibe import Diariser
from scraibe import diarisation, Diariser
@@ -15,7 +15,7 @@ def diariser_instance():
Returns:
Diariser(Obj): An instance of the Diariser class with a mocked token.
"""
with mock.patch.object(Diariser, '_get_token', return_value = 'personal Hugging-Face token')
with mock.patch.object(Diariser, '_get_token', return_value = 'HF_TOKEN' ):
return Diariser('pyannote')