add __array__ method to APRSlicer

AdaptiveParticles · Nov 6, 2023 · 9ce0fde · 9ce0fde
1 parent 12a7c84
commit 9ce0fde
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pyapr/reconstruction/APRSlicer.py b/pyapr/reconstruction/APRSlicer.py
@@ -67,6 +67,10 @@ def shape(self):
     def ndim(self):
         return 3
 
+    def __array__(self):
+        # allows things like np.max(APRSlicer)
+        return np.array(self.parts)
+
     def astype(self, typespec):
         parts = type(type_to_particles(typespec))(np.array(self.parts).astype(typespec))
         dtype = typespec