From 934d6cf21fe713d1661754496f45c0f345ee96e0 Mon Sep 17 00:00:00 2001 From: Brian Kim Date: Wed, 14 Feb 2024 15:42:09 -0800 Subject: [PATCH] lint --- src/allencell_ml_segmenter/prediction/file_input_widget.py | 5 +++-- src/allencell_ml_segmenter/prediction/service.py | 3 ++- src/allencell_ml_segmenter/services/training_service.py | 2 +- 3 files changed, 6 insertions(+), 4 deletions(-) diff --git a/src/allencell_ml_segmenter/prediction/file_input_widget.py b/src/allencell_ml_segmenter/prediction/file_input_widget.py index 628bdc45..49709bb7 100644 --- a/src/allencell_ml_segmenter/prediction/file_input_widget.py +++ b/src/allencell_ml_segmenter/prediction/file_input_widget.py @@ -24,9 +24,10 @@ from allencell_ml_segmenter.prediction.model import ( PredictionModel, PredictionInputMode, - ) -from allencell_ml_segmenter.prediction.service import extract_num_channels_from_image +from allencell_ml_segmenter.prediction.service import ( + extract_num_channels_from_image, +) from allencell_ml_segmenter.widgets.check_box_list_widget import ( CheckBoxListWidget, diff --git a/src/allencell_ml_segmenter/prediction/service.py b/src/allencell_ml_segmenter/prediction/service.py index 74d1e994..756f30f1 100644 --- a/src/allencell_ml_segmenter/prediction/service.py +++ b/src/allencell_ml_segmenter/prediction/service.py @@ -74,6 +74,7 @@ def _determine_input_selection_type(self, path: Path): elif path.suffix == ".csv": return self.extract_num_channels_from_csv(path) + def extract_num_channels_from_image(path: Path): img: AICSImage = AICSImage(str(path)) - return img.dims.C \ No newline at end of file + return img.dims.C diff --git a/src/allencell_ml_segmenter/services/training_service.py b/src/allencell_ml_segmenter/services/training_service.py index ebccd957..219a0d8f 100644 --- a/src/allencell_ml_segmenter/services/training_service.py +++ b/src/allencell_ml_segmenter/services/training_service.py @@ -3,7 +3,7 @@ from allencell_ml_segmenter.core.subscriber import Subscriber from allencell_ml_segmenter.core.event import Event -# from cyto_dl.api.model import CytoDLModel +from cyto_dl.api.model import CytoDLModel # from lightning.pytorch.callbacks import Callback