NDCube fail

DKISTDC · Feb 7, 2025 · 408b9f6 · 408b9f6
1 parent 090a155
commit 408b9f6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/dkist/dataset/tiled_dataset.py b/dkist/dataset/tiled_dataset.py
@@ -227,7 +227,8 @@ def plot(self, slice_index, share_zscale=False, figure=None, swap_tile_limits: L
             figure = plt.gcf()
 
         sliced_dataset = self.slice_tiles[slice_index]
-        if (nd_sliced := len(sliced_dataset.flat[0].shape)) != 2:
+        # This can change to just .shape once we support ndcube >= 2.3
+        if (nd_sliced := len(sliced_dataset.flat[0].data.shape)) != 2:
             raise ValueError(
                 f"Applying slice '{slice_index}' to this dataset resulted in a {nd_sliced} "
                 "dimensional dataset, you should pass a slice which results in a 2D dataset for each tile."