squeeze scale with image (#435)

mehta-lab · Aug 31, 2023 · d644cc2 · d644cc2
1 parent f2d5bb8
commit d644cc2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/recOrder/plugin/main_widget.py b/recOrder/plugin/main_widget.py
@@ -1156,6 +1156,8 @@ def _add_or_update_image_layer(
         """
         if image.shape[0] == 1:
             image = image.squeeze(axis=0)
+            scale = scale[1:]
+
         scale = scale[-image.ndim :]  # match shapes
 
         if name in self.viewer.layers:
@@ -1275,7 +1277,6 @@ def update_overlay_dask_array(self):
             ret_max=self.ret_max,
             cmap=self.colormap,
         )
-
         self._add_or_update_image_layer(
             overlay, self.overlay_name, cmap="rgb", scale=self.overlay_scale
         )