support mean, sum and variance in array_functions

BiocPy · May 24, 2024 · 632a246 · 632a246
1 parent d5fb9e8
commit 632a246
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/delayedarray/DelayedArray.py b/src/delayedarray/DelayedArray.py
@@ -255,6 +255,15 @@ def __array_function__(self, func, types, args, kwargs) -> "DelayedArray":
                 decimals = 0
             return DelayedArray(Round(seed, decimals=decimals))
 
+        if func == numpy.mean:
+            return self.mean(**kwargs)
+
+        if func == numpy.sum:
+            return self.sum(**kwargs)
+
+        if func == numpy.var:
+            return self.var(**kwargs)
+
         if func == numpy.shape:
             return self.shape