diff --git a/tests/test_gui.py b/tests/test_gui.py index 9c6287c..9986e0b 100644 --- a/tests/test_gui.py +++ b/tests/test_gui.py @@ -84,6 +84,7 @@ def test_gui_correct_offset(mw, correct_offset): def test_gui_segm_torch_model(mw, qtbot, monkeypatch): + pytest.importorskip("torch") path = retrieve_data( "fmt-hdf5_cytoshot_full-features_legacy_allev_2023.zip") path_model = retrieve_model( diff --git a/tests/test_gui_manager.py b/tests/test_gui_manager.py index addc4e6..f1c3b9d 100644 --- a/tests/test_gui_manager.py +++ b/tests/test_gui_manager.py @@ -1,5 +1,7 @@ import shutil +import pytest + from dcnum.meta import ppid from chipstream.gui import manager @@ -92,6 +94,7 @@ def test_manager_run_defaults(): def test_manager_run_error_wrong_model(): + pytest.importorskip("torch") model_file = retrieve_model( "segm-torch-model_unet-dcnum-test_g1_910c2.zip")