lint

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,

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
+    return img.dims.C

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