removed unneccesary imports

This commit is contained in:
Tryndaron
2024-04-12 09:37:01 +02:00
parent 9340b83d02
commit 54bedc5376
+2 -12
View File
@@ -1,21 +1,11 @@
import pytest
import torch
from scraibe import Scraibe, Diariser, Transcriber, Transcript, AudioProcessor
from scraibe import Scraibe, Diariser, Transcriber, Transcript
from unittest.mock import MagicMock, patch
import os
import imp
"""
@pytest.fixture
def example_audio_file(audio_test_2.mp4):
audio_path = audio_test_2.mp4
"""
@pytest.fixture
def create_scraibe_instance():
@@ -35,7 +25,7 @@ def test_scraibe_init(create_scraibe_instance):
def test_scraibe_autotranscribe(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)