cleanup

kushalkolar · kushalkolar · commit bfff1138c998 · 2026-04-16T23:45:44.000-04:00
diff --git a/fastplotlib/graphics/features/_image.py b/fastplotlib/graphics/features/_image.py
@@ -48,6 +48,7 @@ def __init__(
         super().__init__(property_name=property_name)
 
         self._colorspace = ColorspacesRGB(colorspace)
+        self._cpu_buffer = cpu_buffer
         data = self._check_data(data, colorspace, cpu_buffer)
 
         self._shape = data.shape
@@ -123,9 +124,6 @@ def __init__(
 
             self.buffer[buffer_index] = texture
 
-        self._colorspace = colorspace
-        self._cpu_buffer = cpu_buffer
-
     @property
     def colorspace(
         self,
diff --git a/fastplotlib/widgets/nd_widget/_video.py b/fastplotlib/widgets/nd_widget/_video.py
@@ -24,4 +24,4 @@ def get_window_output(self, indices: dict[str, Any]):
             return tuple(a.squeeze() for a in windowed_slice)
 
         # convert to numpy array
-        return np.asarray(windowed_slice)
+        return np.asarray(windowed_slice).squeeze()
