Removed extraneous field

brainglobe · Oct 7, 2024 · 211009c · 211009c
1 parent 843da73
commit 211009c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/cellfinder/napari/detect/detect.py b/cellfinder/napari/detect/detect.py
@@ -256,7 +256,6 @@ def widget(
         use_pre_trained_weights: bool,
         weights_only: bool,
         trained_model: Optional[Path],
-        model_weights: Optional[Path],
         batch_size: int,
         misc_options,
         start_plane: int,
@@ -311,9 +310,6 @@ def widget(
         trained_model : Optional[Path]
             Trained model file path (home directory (default) -> pretrained
             weights)
-        model_weights : Optional[Path]
-            Model weights file path (home directory (default) -> pretrained
-            weights)
         start_plane : int
             First plane to process (to process a subset of the data)
         end_plane : int
@@ -379,7 +375,10 @@ def widget(
             max_cluster_size,
         )
 
+        model_weights = None
+
         if weights_only:
+            model_weights = trained_model
             trained_model = None
 
         if use_pre_trained_weights:

diff --git a/cellfinder/napari/detect/detect_containers.py b/cellfinder/napari/detect/detect_containers.py
@@ -131,9 +131,8 @@ def widget_representation(cls) -> dict:
             use_pre_trained_weights=dict(
                 value=cls.defaults()["use_pre_trained_weights"]
             ),
-            weights_only=dict(value=cls.defaults()["weights_only"]),
+            # 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"]
             ),