fix non-finite condition in mean/var calculation (#1381)

2026-09-27 06:18:12 +08:00 · 2020-04-10 12:52:02 -07:00
parent 14021db239
commit 61b330b9b0
1 changed files with 4 additions and 0 deletions
@@ -34,6 +34,10 @@ def _mean_var_n(X):
    if fp_err_occurred:
        mean[np.isfinite(mean) == False] = 0  # noqa: E712
        v[np.isfinite(v) == False] = 0  # noqa: E712
+    else:
+        mean[np.isnan(mean)] = 0
+        v[np.isnan(v)] = 0
+
    return mean, v, n