Merge pull request #7 from weigertlab/fix-ilp-err-msg

Add gentle error for missing ILP install

Author: bentaculum

pyproject.toml

@@ -39,6 +39,13 @@ testing = [
     "napari",
     "pyqt5",
 ]
+dev = [
+    "napari-trackastra[testing]",
+    "pre-commit",
+    "ruff",
+    "twine",
+    "build",
+]
 
 [project.entry-points."napari.manifest"]
 napari-trackastra = "napari_trackastra:napari.yaml"

src/napari_trackastra/_widget.py

@@ -130,6 +130,7 @@ def __init__(
         self._model_type.changed.connect(self._model_type_changed)
         self._model_pretrained.changed.connect(self._update_model)
         self._model_path.changed.connect(self._update_model)
+        self._linking_mode.changed.connect(self._update_linking_mode)
         self._run_button.changed.connect(self._run)
 
         # self._save_path.changed.connect(self._save)
@@ -171,6 +172,20 @@ def _update_model(self, event=None):
                 self._model_path.value, device=self._device
             )
 
+    def _update_linking_mode(self, event=None):
+        if self._linking_mode.value == "ilp":
+            try:
+                import motile  # noqa F401
+
+                return True
+            except ModuleNotFoundError:
+                napari.utils.notifications.show_warning(
+                    "For tracking with an ILP, please conda install the optional `motile`"
+                    " dependency following https://funkelab.github.io/motile/install.html."
+                )
+                return False
+        return True
+
     def _show_activity_dock(self, state=True):
         # show/hide activity dock if there is actual progress to see
         self._viewer.window._status_bar._toggle_activity_dock(state)
@@ -181,6 +196,9 @@ def _run(self, event=None):
         if self.model is None:
             raise ValueError("Model not loaded")
 
+        if not self._update_linking_mode():
+            return
+
         imgs = np.asarray(self._image_layer.value.data)
         masks = np.asarray(self._mask_layer.value.data)