finished test fixes

src/allencell_ml_segmenter/_tests/prediction/test_file_input_widget.py

@@ -117,7 +117,7 @@ def test_populate_input_channel_combobox(qtbot: QtBot) -> None:
     # Arrange
     prediction_model: PredictionModel = PredictionModel()
     prediction_file_input: PredictionFileInput = PredictionFileInput(
-        prediction_model
+        prediction_model, viewer=FakeViewer()
     )
     prediction_model.set_max_channels(6)
 

src/allencell_ml_segmenter/_tests/test_files/csv/test_csv.csv

@@ -1,2 +1,4 @@
-to_predict
-./src/allencell_ml_segmenter/_tests/test_files/images/test_2_channels.tiff
+,raw,split
+0,./src/allencell_ml_segmenter/_tests/test_files/images/test_2_channels.tiff,test
+1,./src/allencell_ml_segmenter/_tests/test_files/images/test_2_channels.tiff,test
+2,./src/allencell_ml_segmenter/_tests/test_files/images/test_2_channels.tiff,test

src/allencell_ml_segmenter/prediction/service.py

@@ -56,8 +56,7 @@ def extract_num_channels_in_folder(self, path: Path) -> int:
         # ignore hidden files
         while str(first_image.name).startswith("."):
             first_image = next(path_generator)
-
-        extract_num_channels_from_image(str(first_image.resolve()))
+        return extract_num_channels_from_image(str(first_image.resolve()))
 
     def extract_num_channels_from_csv(self, path: Path):
         with open(path) as file:

src/allencell_ml_segmenter/services/prediction_service.py

@@ -166,7 +166,7 @@ def build_overrides(
     def write_csv_for_inputs(self, list_images: List[Path]) -> None:
         data_folder: Path = self._experiments_model.get_csv_path()
         data_folder.mkdir(parents=False, exist_ok=True)
-        csv_path: Path = data_folder / "prediction_input.csv"
+        csv_path: Path = data_folder / "test_csv.csv"
         with open(csv_path, "w") as file:
             writer: csv.writer = csv.writer(file)
             writer.writerow(["", "raw", "split"])