Skip to content

Commit 75e3466

Browse files
committed
fix
1 parent f6d9deb commit 75e3466

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

flox/aggregate_flox.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -118,5 +118,7 @@ def nanmean(group_idx, array, *, axis=-1, size=None, fill_value=None, dtype=None
118118
masked = np.where(mask, 0, array)
119119
out = sum(group_idx, masked, size=size, axis=axis, dtype=dtype, fill_value=fill_value)
120120
with np.errstate(invalid="ignore", divide="ignore"):
121-
out /= sum(group_idx, mask.view(np.int8), size=size, axis=axis, fill_value=0, dtype=np.intp)
121+
out /= sum(
122+
group_idx, (~mask).view(np.int8), size=size, axis=axis, fill_value=0, dtype=np.intp
123+
)
122124
return out

0 commit comments

Comments
 (0)