From 7f2260bade7cc66f466de8f123c714ae938a1ccb Mon Sep 17 00:00:00 2001 From: yrkim98 Date: Wed, 11 Oct 2023 10:57:57 -0700 Subject: [PATCH] pr --- src/allencell_ml_segmenter/curation/main_view.py | 1 + src/allencell_ml_segmenter/services/training_service.py | 4 ++-- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/src/allencell_ml_segmenter/curation/main_view.py b/src/allencell_ml_segmenter/curation/main_view.py index ebec07c8..9fa9678e 100644 --- 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 index 3f7ed828..e876a620 100644 --- 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: """