Cleanup + tests

- Removed previous train script
- Fix tests
- Enable test workflow on GH

Author: C-Achard

.github/workflows/test_and_deploy.yml

@@ -7,6 +7,7 @@ on:
   push:
     branches:
       - main
+      - cy/wnet-train
     tags:
       - "v*" # Push events to matching v*, i.e. v1.0, v20.15.10
   pull_request:

napari_cellseg3d/_tests/test_models.py

@@ -115,7 +115,7 @@ def test_pretrained_weights_compatibility():
     for model_name in MODEL_LIST:
         file_name = MODEL_LIST[model_name].weights_file
         WeightsDownloader().download_weights(model_name, file_name)
-        model = MODEL_LIST[model_name](input_img_size=[128, 128, 128])
+        model = MODEL_LIST[model_name](input_img_size=[64, 64, 64])
         try:
             model.load_state_dict(
                 torch.load(

napari_cellseg3d/_tests/test_unsup_training.py

@@ -5,6 +5,7 @@
     Trainer,
 )
 
+
 def test_unsupervised_worker(make_napari_viewer_proxy):
     im_path = Path(__file__).resolve().parent / "res/test.tif"
     # im_path_str = str(im_path)
@@ -34,8 +35,8 @@ def test_unsupervised_worker(make_napari_viewer_proxy):
     assert eval_dataloader is None
     assert data_shape == (6, 6, 6)
 
-    widget.images_filepaths = [str(im_path.parent)]
-    widget.labels_filepaths = [str(im_path.parent)]
+    widget.images_filepaths = [str(im_path)]
+    widget.labels_filepaths = [str(im_path)]
     # widget.unsupervised_eval_data = widget.create_train_dataset_dict()
     worker = widget._create_worker(additional_results_description="TEST_3")
     dataloader, eval_dataloader, data_shape = worker._get_data()

napari_cellseg3d/_tests/test_wnet_training.py

@@ -1,25 +0,0 @@
-#######################################################
-# Disabled as it takes too much memory for GH actions #
-#######################################################
-
-
-# from pathlib import Path
-# from napari_cellseg3d.code_models.models.wnet import train_wnet as t
-#
-# def test_wnet_training():
-#     config = t.Config()
-#
-#     config.batch_size = 1
-#     config.num_epochs = 1
-#
-#     config.train_volume_directory = str(Path(__file__).resolve().parent / "res/wnet_test")
-#     config.eval_volume_directory = config.train_volume_directory
-#     config.save_every = 1
-#     config.val_interval = 2  # skip validation
-#     config.save_model_path = config.train_volume_directory + "/test.pth"
-#
-#     ncuts_loss, rec_loss, model = t.train(train_config=config)
-#
-#     assert ncuts_loss is not None
-#     assert rec_loss is not None
-#     assert model is not None