Merge branch 'array_interface'

tskisner · Oct 25, 2022 · 5c434b3 · 5c434b3
2 parents ef01d9d + 3e83e70
commit 5c434b3
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/pshmem/shmem.py b/pshmem/shmem.py
@@ -332,9 +332,14 @@ def __iter__(self):
         else:
             return iter(self.data)
 
-    def __array__(self):
+    def __array__(self, *args, **kwargs):
+        """Provide the underlying numpy data view as the array."""
+        return self.data.__array__(*args, **kwargs)
+
+    @property
+    def __array_interface__(self):
         """Provide the underlying numpy data view as the array interface."""
-        return self.data
+        return self.data.__array_interface__
 
     def __repr__(self):
         val = "<MPIShared"