fix: execstack

chidiwilliams · Jan 5, 2024 · 83b789d · 83b789d
1 parent 47fbcec
commit 83b789d
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/tests/widgets/preferences_dialog/models_preferences_widget_test.py b/tests/widgets/preferences_dialog/models_preferences_widget_test.py
@@ -55,11 +55,7 @@ def test_should_download_model(self, qtbot: QtBot, clear_model_cache):
         )
         qtbot.add_widget(widget)
 
-        model = TranscriptionModel(
-            model_type=ModelType.WHISPER, whisper_model_size=WhisperModelSize.TINY
-        )
-
-        assert model.get_local_model_path() is None
+        assert widget.model.get_local_model_path() is None
 
         available_item = widget.model_list_widget.topLevelItem(1)
         assert available_item.text(0) == "Available for Download"
@@ -87,8 +83,7 @@ def downloaded_model():
                 or _available_item.child(0).text(0) != "Tiny"
             )
 
-            # model file exists
-            assert os.path.isfile(get_whisper_file_path(size=model.whisper_model_size))
+            assert os.path.isfile(widget.model.get_local_model_path())
 
         qtbot.wait_until(callback=downloaded_model, timeout=60_000)