Fix formula summing for change in Numpy sum

nipy · Jan 2, 2023 · aaae2a3 · aaae2a3
1 parent 9d1f8d8
commit aaae2a3
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/nipy/algorithms/statistics/formula/formulae.py b/nipy/algorithms/statistics/formula/formulae.py
@@ -569,9 +569,10 @@ def fromrec(rec, keep=[], drop=[]):
         for d in drop:
             del(f[d])
         if keep:
-            return np.sum([t for n, t in f.items() if n in keep])
+            elements = [t for n, t in f.items() if n in keep]
         else:
-            return np.sum(list(f.values()))
+            elements = f.values()
+        return sum(elements, start=Formula([]))
 
     def subs(self, old, new):
         """ Perform a sympy substitution on all terms in the Formula