Update worker_training.py

AdaptiveMotorControlLab · Dec 21, 2024 · f722137 · f722137
1 parent a76037c
commit f722137
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/napari_cellseg3d/code_models/worker_training.py b/napari_cellseg3d/code_models/worker_training.py
@@ -281,7 +281,11 @@ def get_dataset(self, train_transforms):
         load_single_images = Compose(
             [
                 LoadImaged(keys=["image"]),
-                EnsureChannelFirstd(keys=["image"], channel_dim="no_channel", strict_check=False),
+                EnsureChannelFirstd(
+                    keys=["image"],
+                    channel_dim="no_channel",
+                    strict_check=False,
+                ),
                 Orientationd(keys=["image"], axcodes="PLI"),
                 SpatialPadd(
                     keys=["image"],