From 0a28d67f4546549e9cfcb8690f4e587ffbcf4bf2 Mon Sep 17 00:00:00 2001
From: IgorTatarnikov <igor.tatarnikov@gmail.com>
Date: Mon, 7 Oct 2024 21:41:32 +0100
Subject: [PATCH] Fixed tests v3

---
 cellfinder/napari/detect/detect.py            | 2 +-
 cellfinder/napari/detect/detect_containers.py | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/cellfinder/napari/detect/detect.py b/cellfinder/napari/detect/detect.py
index 66bbbb22..07adaf0b 100644
--- 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
index 28f45961..4dfda0c2 100644
--- 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"]
             ),