Make NDCube.meta have same shape as NDCube if meta is axis-aware.

sunpy · Jul 2, 2024 · 6d38c67 · 6d38c67
1 parent 2373209
commit 6d38c67
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/ndcube/meta.py b/ndcube/meta.py
@@ -5,7 +5,7 @@
 
 import numpy as np
 
-__all__ = ["NDMeta"]
+__all__ = ["NDMeta", "NDMetaABC"]
 
 
 class NDMetaABC(collections.abc.Mapping):

diff --git a/ndcube/ndcube.py b/ndcube/ndcube.py
@@ -27,6 +27,7 @@
 from ndcube import utils
 from ndcube.extra_coords.extra_coords import ExtraCoords, ExtraCoordsABC
 from ndcube.global_coords import GlobalCoords, GlobalCoordsABC
+from ndcube.meta import NDMetaABC
 from ndcube.mixins import NDCubeSlicingMixin
 from ndcube.ndcube_sequence import NDCubeSequence
 from ndcube.utils.exceptions import warn_deprecated, warn_user
@@ -399,6 +400,10 @@ def __init__(self, data, wcs=None, uncertainty=None, mask=None, meta=None,
                 global_coords = deepcopy(global_coords)
             self._global_coords = global_coords
 
+        # If meta is axis-aware, make it to have same shape as cube.
+        if isinstance(self.meta, NDMetaABC):
+            self.meta.data_shape = self.shape
+
     @property
     def extra_coords(self):
         # Docstring in NDCubeABC.