lint with updated black

src/allencell_ml_segmenter/_tests/services/test_prediction_service.py

@@ -125,11 +125,11 @@ def test_build_overrides() -> None:
     prediction_model.set_image_input_channel_index(3)
 
     # act
-    overrides: Dict[
-        str, Union[str, int, float, bool]
-    ] = prediction_service.build_overrides(
-        experiments_model.get_experiment_name(),
-        experiments_model.get_checkpoint(),
+    overrides: Dict[str, Union[str, int, float, bool]] = (
+        prediction_service.build_overrides(
+            experiments_model.get_experiment_name(),
+            experiments_model.get_checkpoint(),
+        )
     )
 
     # assert
@@ -183,11 +183,11 @@ def test_build_overrides_experiment_none() -> None:
     # act/assert
     # Experiment name is None, so build_overrides should throw a ValueError
     with pytest.raises(ValueError):
-        overrides: Dict[
-            str, Union[str, int, float, bool]
-        ] = prediction_service.build_overrides(
-            experiments_model.get_experiment_name(),
-            experiments_model.get_checkpoint(),
+        overrides: Dict[str, Union[str, int, float, bool]] = (
+            prediction_service.build_overrides(
+                experiments_model.get_experiment_name(),
+                experiments_model.get_checkpoint(),
+            )
         )
 
 
@@ -217,9 +217,9 @@ def test_build_overrides_checkpoint_none() -> None:
     # act/assert
     # Checkpoint is None, so build_overrides should throw a ValueError
     with pytest.raises(ValueError):
-        overrides: Dict[
-            str, Union[str, int, float, bool]
-        ] = prediction_service.build_overrides(
-            experiments_model.get_experiment_name(),
-            experiments_model.get_checkpoint(),
+        overrides: Dict[str, Union[str, int, float, bool]] = (
+            prediction_service.build_overrides(
+                experiments_model.get_experiment_name(),
+                experiments_model.get_checkpoint(),
+            )
         )

src/allencell_ml_segmenter/curation/curation_service.py

@@ -395,9 +395,7 @@ def update_curation_record(self, use_image: bool) -> None:
         Update the curation record with the users selection for the current image
         """
         # DEAL WITH EXCLUDING MASKS
-        excluding_mask_path: Union[
-            Path, str
-        ] = (
+        excluding_mask_path: Union[Path, str] = (
             self._curation_model.get_current_excluding_mask_path_and_reset_mask()
         )
         if excluding_mask_path is not None:
@@ -406,9 +404,9 @@ def update_curation_record(self, use_image: bool) -> None:
             excluding_mask_path = ""
 
         # DEAL WITH MERGING MASKS
-        merging_mask_path: Union[
-            Path, str
-        ] = self._curation_model.get_current_merging_mask_path()
+        merging_mask_path: Union[Path, str] = (
+            self._curation_model.get_current_merging_mask_path()
+        )
         if merging_mask_path is not None:
             # user has drawn and saved merging masks.
             merging_mask_path = str(merging_mask_path)

src/allencell_ml_segmenter/main/i_experiments_model.py

@@ -5,7 +5,6 @@
 
 
 class IExperimentsModel(Publisher):
-
     """
     Interface for implementing and testing ExperimentsModel
     """

src/allencell_ml_segmenter/napari/napari_reader.py

@@ -6,6 +6,7 @@
 implement multiple readers or even other plugin contributions. see:
 https://napari.org/stable/plugins/guides.html?#readers
 """
+
 import numpy as np
 from typing import Union, Callable, List, Tuple, Dict
 

src/allencell_ml_segmenter/napari/napari_writer.py

@@ -7,6 +7,7 @@
 
 Replace code below according to your needs.
 """
+
 from __future__ import annotations
 
 from typing import TYPE_CHECKING, Any, List, Sequence, Tuple, Union

src/allencell_ml_segmenter/napari/sample_data.py

@@ -7,6 +7,7 @@
 
 Replace code below according to your needs.
 """
+
 from __future__ import annotations
 from typing import List, Tuple, Dict
 

src/allencell_ml_segmenter/services/prediction_service.py

@@ -112,8 +112,8 @@ def build_overrides(
         # if channel is not set, will default to same channel used to train
         channel: int = self._prediction_model.get_image_input_channel_index()
         if channel:
-            overrides[
-                "data.transforms.predict.transforms[0].reader[0].C"
-            ] = channel
+            overrides["data.transforms.predict.transforms[0].reader[0].C"] = (
+                channel
+            )
 
         return overrides

src/allencell_ml_segmenter/training/training_model.py

@@ -54,9 +54,9 @@ def __init__(
         self._hardware_type: Hardware = None
         self._images_directory: Path = None
         self._channel_index: Union[int, None] = None
-        self._model_path: Union[
-            Path, None
-        ] = None  # if None, start a new model
+        self._model_path: Union[Path, None] = (
+            None  # if None, start a new model
+        )
         self._patch_size: PatchSize = None
         self._spatial_dims: int = None
         self._max_epoch: int = None