Fix line too long

AustralianCancerDataNetwork · Nov 24, 2024 · 00e4076 · 00e4076
1 parent ff3cfdd
commit 00e4076
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/pydicer/visualise/data.py b/pydicer/visualise/data.py
@@ -204,7 +204,10 @@ def visualise(
                         for f in struct_dir.glob("*.nii.gz")
                     }
 
-                    masks = {k: masks[k] for k in masks if masks[k].GetPixelID() in render_mask_types}
+                    masks = {
+                        k: masks[k] for k in masks
+                        if masks[k].GetPixelID() in render_mask_types
+                    }
 
                     if len(masks) == 0:
                         logger.warning(
@@ -283,7 +286,10 @@ def visualise(
                             f.name.replace(".nii.gz", ""): sitk.ReadImage(str(f))
                             for f in struct_dir.glob("*.nii.gz")
                         }
-                        masks = {k: masks[k] for k in masks if masks[k].GetPixelID() in render_mask_types}
+                        masks = {
+                            k: masks[k] for k in masks
+                            if masks[k].GetPixelID() in render_mask_types
+                        }
 
                         if len(masks) == 0:
                             logger.warning(