Update dimension.py

pycroscopy · Dec 15, 2024 · 2199616 · 2199616
1 parent d7c0593
commit 2199616
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/sidpy/sid/dimension.py b/sidpy/sid/dimension.py
@@ -102,10 +102,12 @@ def __array_finalize__(self, obj):
 
     def __array_wrap__(self, out_arr, context=None):
         # just call the parent
-        super(Dimension, self).values.__array_wrap__(self, out_arr, context)
+        values = super(Dimension, self).__array_wrap__(self, out_arr, context)
+        new_dim = Dimension(values, name=self.name, quantity=self.quantity, units=self.units)
         # return correct values
-        return out_arr
-
+        return new_dim
+
+
     def __repr__(self):
         return '{}:  {} ({}) of size {}'.format(self.name, self.quantity, self.units, self.shape)