adjusted tests after merge

HelmholtzAI-Consultants-Munich · Jan 25, 2024 · ee70715 · ee70715
1 parent 06ffbd2
commit ee70715
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/server/test/configs/test_config_fcnn.cfg b/src/server/test/configs/test_config_fcnn.cfg
@@ -21,6 +21,7 @@
             "model_class": "FCNN", 
             "in_channels": 1,
             "num_classes": 3,
+            "features":[64,128,256,512],
             "black_bg": "False",
             "include_mask": "False"
         }

diff --git a/src/server/test/test_models.py b/src/server/test/test_models.py
@@ -22,7 +22,7 @@ def test_update_configs():
     train_config = read_config('train', config_path='test/configs/test_config_RF.cfg')
     eval_config = read_config('eval', config_path='test/configs/test_config_RF.cfg')
 
-    model = models.CustomCellposeModel(model_config,train_config,eval_config)
+    model = models.CustomCellposeModel(model_config,train_config,eval_config, "Cellpose")
 
     new_train_config = {"param1": "value1"}
     new_eval_config = {"param2": "value2"}