comments

AllenCell · Feb 6, 2024 · eef2ba8 · eef2ba8
1 parent 6c2fb9b
commit eef2ba8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/allencell_ml_segmenter/services/prediction_service.py b/src/allencell_ml_segmenter/services/prediction_service.py
@@ -74,7 +74,7 @@ def _predict_model(self, _: Event) -> None:
             )
             continue_prediction = False
 
-        # Set input images
+        # Create a CSV if user selects a folder of input images.
         if (
             self._prediction_model.get_prediction_input_mode()
             == PredictionInputMode.FROM_PATH
@@ -129,6 +129,7 @@ def build_overrides(
                 "data.transforms.predict.transforms[0].reader[0].C"
             ] = channel
 
+        # Need these overrides to load in csv's
         overrides["data.columns"] = ["raw", "split"]
         overrides["data.split_column"] = "split"