a slice of an array of a given type is still of that same type

Jammy2211 · Dec 18, 2023 · 7cb6cc1 · 7cb6cc1
1 parent a1669cf
commit 7cb6cc1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/autoarray/abstract_ndarray.py b/autoarray/abstract_ndarray.py
@@ -239,7 +239,10 @@ def reshape(self, *args, **kwargs):
         return self._array.reshape(*args, **kwargs)
 
     def __getitem__(self, item):
-        return self._array[item]
+        result = self._array[item]
+        if isinstance(item, slice):
+            return self.with_new_array(result)
+        return result
 
     def __setitem__(self, key, value):
         if isinstance(key, (np.ndarray, AbstractNDArray, Array)):