fix: correctly check if time dim is needed in preComputedMinMaxValues

SchmollerLab · Nov 20, 2024 · 4395b6e · 4395b6e
1 parent 058560f
commit 4395b6e
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/cellacdc/widgets.py b/cellacdc/widgets.py
@@ -6313,9 +6313,13 @@ def preComputedMinMaxValues(self, data: dict):
         self.minMaxValuesMapper = {}
         for pos_i, posData in enumerate(data):
             img_data = posData.img_data
-            if posData.img_data.ndim == 2:
+            requires_time_dim = (
+                posData.img_data.ndim == 2
+                or (posData.img_data.ndim == 3 and posData.SizeZ > 1)
+            )
+            if requires_time_dim:
                 img_data = (img_data,)
-                
+
             for frame_i, image in enumerate(img_data):
                 if image.ndim == 2:
                     image = (image,)