Fix default LR + sup. test

Author: C-Achard

napari_cellseg3d/_tests/test_supervised_training.py

@@ -12,12 +12,15 @@
 im_path = Path(__file__).resolve().parent / "res/test.tif"
 im_path_str = str(im_path)
 
-def test_create_supervised_worker_from_config(make_napari_viewer_proxy):
 
+def test_create_supervised_worker_from_config(make_napari_viewer_proxy):
     viewer = make_napari_viewer_proxy()
     widget = Trainer(viewer=viewer)
     widget.device_choice.setCurrentIndex(0)
-    worker = widget._create_worker()
+    widget.model_choice.setCurrentIndex(0)
+    widget._toggle_unsupervised_mode(enabled=False)
+    assert widget.model_choice.currentText() == list(MODEL_LIST.keys())[0]
+    worker = widget._create_worker(additional_results_description="test")
     default_config = config.SupervisedTrainingWorkerConfig()
     excluded = [
         "results_path_folder",

napari_cellseg3d/code_plugins/plugin_model_training.py

@@ -443,15 +443,15 @@ def _toggle_unsupervised_mode(self, enabled=False):
             self.start_btn = self.start_button_unsupervised
             self.image_filewidget.text_field.setText("Validation images")
             self.labels_filewidget.text_field.setText("Validation labels")
-            self.learning_rate_choice.lr_value_choice.setValue(1)
-            self.learning_rate_choice.lr_exponent_choice.setCurrentIndex(1)
+            self.learning_rate_choice.lr_value_choice.setValue(2)
+            self.learning_rate_choice.lr_exponent_choice.setCurrentIndex(3)
         else:
             unsupervised = False
             self.start_btn = self.start_button_supervised
             self.image_filewidget.text_field.setText("Images directory")
             self.labels_filewidget.text_field.setText("Labels directory")
-            self.learning_rate_choice.lr_value_choice.setValue(2)
-            self.learning_rate_choice.lr_exponent_choice.setCurrentIndex(3)
+            self.learning_rate_choice.lr_value_choice.setValue(1)
+            self.learning_rate_choice.lr_exponent_choice.setCurrentIndex(1)
 
         supervised = not unsupervised
         self.unsupervised_mode = unsupervised