Update worker_training.py

Author: C-Achard

napari_cellseg3d/code_models/worker_training.py

@@ -201,7 +201,11 @@ def get_patch_dataset(self, train_transforms):
         patch_func = Compose(
             [
                 LoadImaged(keys=["image"], image_only=True),
-                EnsureChannelFirstd(keys=["image"], channel_dim="no_channel", strict_check=False),
+                EnsureChannelFirstd(
+                    keys=["image"],
+                    channel_dim="no_channel",
+                    strict_check=False,
+                ),
                 RandSpatialCropSamplesd(
                     keys=["image"],
                     roi_size=(
@@ -236,7 +240,9 @@ def get_dataset_eval(self, eval_dataset_dict):
             [
                 LoadImaged(keys=["image", "label"]),
                 EnsureChannelFirstd(
-                    keys=["image", "label"], channel_dim="no_channel", strict_check=False
+                    keys=["image", "label"],
+                    channel_dim="no_channel",
+                    strict_check=False,
                 ),
                 # RandSpatialCropSamplesd(
                 #     keys=["image", "label"],