admin/scribe
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

removed unneccesary imports

Browse Source
This commit is contained in:
Tryndaron
2024-04-12 09:37:01 +02:00
parent 9340b83d02
commit 54bedc5376
1 changed files with 2 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/test_autotranscript.py
+2 -12
View File
@@ -1,21 +1,11 @@
import pytest import pytest
import torch from scraibe import Scraibe, Diariser, Transcriber, Transcript
from scraibe import Scraibe, Diariser, Transcriber, Transcript, AudioProcessor
from unittest.mock import MagicMock, patch from unittest.mock import MagicMock, patch
import os import os
import imp
"""
@pytest.fixture
def example_audio_file(audio_test_2.mp4):
audio_path = audio_test_2.mp4
"""
@pytest.fixture @pytest.fixture
def create_scraibe_instance(): def create_scraibe_instance():
@@ -35,7 +25,7 @@ def test_scraibe_init(create_scraibe_instance):
def test_scraibe_autotranscribe(create_scraibe_instance): def test_scraibe_autotranscribe(create_scraibe_instance):
model = create_scraibe_instance model = create_scraibe_instance
transcript = model.autotranscribe('test/audio_test_2.mp4') transcript = model.autotranscribe('test/audio_test.mp4')
assert isinstance(transcript, Transcript) assert isinstance(transcript, Transcript)