fix: execstack

chidiwilliams · Jan 5, 2024 · b6147f0 · b6147f0
1 parent 4d6034e
commit b6147f0
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/tests/widgets/transcription_task_folder_watcher_test.py b/tests/widgets/transcription_task_folder_watcher_test.py
@@ -4,7 +4,7 @@
 
 from pytestqt.qtbot import QtBot
 
-from buzz.model_loader import TranscriptionModel
+from buzz.model_loader import TranscriptionModel, ModelType
 from buzz.transcriber import (
     Task,
     DEFAULT_WHISPER_TEMPERATURE,
@@ -24,6 +24,12 @@
 
 
 class TestTranscriptionTaskFolderWatcher:
+    def default_model(self):
+        model_type = next(
+            model_type for model_type in ModelType if model_type.is_available()
+        )
+        return TranscriptionModel(model_type=model_type)
+
     def test_should_add_task_not_in_tasks(self, qtbot: QtBot):
         input_directory = mkdtemp()
         watcher = TranscriptionTaskFolderWatcher(
@@ -35,7 +41,7 @@ def test_should_add_task_not_in_tasks(self, qtbot: QtBot):
                 file_transcription_options=FileTranscriptionPreferences(
                     language=None,
                     task=Task.TRANSCRIBE,
-                    model=TranscriptionModel(),
+                    model=self.default_model(),
                     word_level_timings=False,
                     temperature=DEFAULT_WHISPER_TEMPERATURE,
                     initial_prompt="",