Fixed tests v3

brainglobe · Oct 7, 2024 · 0a28d67 · 0a28d67
1 parent fb6b8d3
commit 0a28d67
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/cellfinder/napari/detect/detect.py b/cellfinder/napari/detect/detect.py
@@ -256,7 +256,7 @@ def widget(
         use_pre_trained_weights: bool,
         weights_only: bool,
         trained_model: Optional[Path],
-        model_weights,
+        model_weights: Optional[Path],
         batch_size: int,
         misc_options,
         start_plane: int,

diff --git a/cellfinder/napari/detect/detect_containers.py b/cellfinder/napari/detect/detect_containers.py
@@ -133,6 +133,7 @@ def widget_representation(cls) -> dict:
             ),
             weights_only=dict(value=cls.defaults()["weights_only"]),
             trained_model=dict(value=cls.defaults()["trained_model"]),
+            model_weights=dict(value=cls.defaults()["model_weights"]),
             skip_classification=dict(
                 value=cls.defaults()["skip_classification"]
             ),