pr

AllenCell · Oct 11, 2023 · 7f2260b · 7f2260b
1 parent e53d632
commit 7f2260b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/allencell_ml_segmenter/curation/main_view.py b/src/allencell_ml_segmenter/curation/main_view.py
@@ -259,5 +259,6 @@ def save_curation_record(self, path: Path):
                     )
                 f.flush()
 
+        # TODO: WRITE ACTUAL VALIDATION AND TEST SETS
         shutil.copy(path, parent_path / "valid.csv")
         shutil.copy(path, parent_path / "test.csv")
diff --git a/src/allencell_ml_segmenter/services/training_service.py b/src/allencell_ml_segmenter/services/training_service.py
@@ -4,7 +4,7 @@
 # from lightning.pytorch.callbacks import Callback
 
 # disabled for tests (cant import in ci yet)
-from cyto_dl.train import main as cyto_train
+# from cyto_dl.train import main as cyto_train
 
 import sys
 from allencell_ml_segmenter.main.experiments_model import ExperimentsModel
@@ -95,7 +95,7 @@ def train_model_handler(self, _: Event) -> None:
             self._set_config_dir()
 
             # disabled for tests (cant import in ci yet)
-            cyto_train()
+            # cyto_train()
 
     def _set_experiment(self) -> None:
         """