env hf token

This commit is contained in:
Tryndaron
2024-04-03 16:10:15 +02:00
parent 2b4819c68f
commit 018a2bd1cf
+6 -2
View File
@@ -2,7 +2,7 @@ import pytest
import torch import torch
from scraibe import Scraibe, Diariser, Transcriber, Transcript, AudioProcessor from scraibe import Scraibe, Diariser, Transcriber, Transcript, AudioProcessor
from unittest.mock import MagicMock, patch from unittest.mock import MagicMock, patch
import os
""" """
@pytest.fixture @pytest.fixture
@@ -13,7 +13,11 @@ def example_audio_file(audio_test_2.mp4):
@pytest.fixture @pytest.fixture
def create_scraibe_instance(): def create_scraibe_instance():
return Scraibe() if "HF_TOKEN" in os.environ:
return Scraibe(use_auth_token=os.environ["HF_TOKEN"] )
else:
return Scraibe()