We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 07f7007 commit e319529Copy full SHA for e319529
1 file changed
tests/test_funcs.py
@@ -1623,7 +1623,8 @@ def test_against_median(
1623
a_np[rng.random(n) < rng.random(n) * 0.5] = np.nan
1624
if w_np is not None:
1625
# ensure at least one NaN on non-null weight:
1626
- a_np[w_np > 0][0] = np.nan
+ nz_weights_idx, = np.where(w_np > 0)
1627
+ a_np[nz_weights_idx[0]] = np.nan
1628
m = "averaged_inverted_cdf"
1629
1630
np_median = np.nanmedian if nan_policy == "omit" else np.median
0 commit comments