Add single if statement for warn

kbressem · kbressem · commit f4992bd42621 · 2023-03-10T17:53:58.000-05:00
Avoids crashing of the function if `ratios_[i] = 0` cannot be updated
diff --git a/monai/transforms/utils.py b/monai/transforms/utils.py
@@ -569,8 +569,9 @@ def generate_label_classes_crop_centers(
         raise ValueError(f"ratios should not contain negative number, got {ratios_}.")
 
     for i, array in enumerate(indices):
-        if len(array) == 0 and warn:
-            warnings.warn(f"no available indices of class {i} to crop, set the crop ratio of this class to zero.")
+        if len(array) == 0:
+            if warn:
+                warnings.warn(f"no available indices of class {i} to crop, set the crop ratio of this class to zero.")
             ratios_[i] = 0
 
     centers = []
