Resolved copying issue with units

cedadev · Aug 13, 2024 · 358a0ce · 358a0ce
1 parent a3d141d
commit 358a0ce
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/CFAPyX/partition.py b/CFAPyX/partition.py
@@ -184,6 +184,7 @@ def copy(self, newextent=None):
             kwargs['extent'] = self._combine_slices(newextent)
 
         new_instance = SuperLazyArrayLike(
+            self.shape,
             **kwargs
             )
         return new_instance

diff --git a/CFAPyX/wrappers.py b/CFAPyX/wrappers.py
@@ -504,10 +504,6 @@ def __init__(self,
 
         :param aggregated_calendar:     None
         """
-        if 'units' in kwargs:
-            if not aggregated_units:
-                aggregated_units = kwargs['units']
-            kwargs.pop('units')
 
         super().__init__(filename, address, units=aggregated_units, **kwargs)
         self.aggregated_units    = aggregated_units
@@ -520,6 +516,12 @@ def copy(self, extent=None):
         """
 
         kwargs = self.get_kwargs()
+
+        if 'units' in kwargs:
+            if not kwargs['aggregated_units']:
+                kwargs['aggregated_units'] = kwargs['units']
+            kwargs.pop('units')
+
         if extent:
             kwargs['extent'] = self._combine_slices(extent)