Skip to content

Commit 20e1cd4

Browse files
authored
Silence more warnings, engine='flox' (#198)
1 parent 8b1434d commit 20e1cd4

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

flox/aggregate_flox.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -107,13 +107,15 @@ def mean(group_idx, array, *, axis=-1, size=None, fill_value=None, dtype=None):
107107
if fill_value is None:
108108
fill_value = 0
109109
out = sum(group_idx, array, axis=axis, size=size, dtype=dtype, fill_value=fill_value)
110-
out /= nanlen(group_idx, array, size=size, axis=axis, fill_value=0)
110+
with np.errstate(invalid="ignore", divide="ignore"):
111+
out /= nanlen(group_idx, array, size=size, axis=axis, fill_value=0)
111112
return out
112113

113114

114115
def nanmean(group_idx, array, *, axis=-1, size=None, fill_value=None, dtype=None):
115116
if fill_value is None:
116117
fill_value = 0
117118
out = nansum(group_idx, array, size=size, axis=axis, dtype=dtype, fill_value=fill_value)
118-
out /= nanlen(group_idx, array, size=size, axis=axis, fill_value=0)
119+
with np.errstate(invalid="ignore", divide="ignore"):
120+
out /= nanlen(group_idx, array, size=size, axis=axis, fill_value=0)
119121
return out

0 commit comments

Comments
 (0)