Closed
Description
import numpy as np
import numpy_groupies as npg
s = npg.aggregate_numpy.aggregate(
[0, 0, 0, 1, 1, 0, 0], [2, 2, np.nan, 2, 2, 2, 2], func="nancumsum"
)
ss = npg.aggregate_numba.aggregate(
[0, 0, 0, 1, 1, 0, 0], [2, 2, np.nan, 2, 2, 2, 2], func="nancumsum"
)
np.testing.assert_allclose(s, ss)
AssertionError:
Not equal to tolerance rtol=1e-07, atol=0
(shapes (6,), (7,) mismatch)
x: array([2., 4., 2., 4., 6., 8.])
y: array([2., 4., 4., 2., 4., 6., 8.])
npg.__version__
Out[5]: '0.9.22'
Doesn't look like the nancumsum function is being chosen.
numpy-groupies/numpy_groupies/aggregate_numpy.py
Lines 256 to 261 in 9fc51ad