From 61b330b9b0a4fc55e6ddc61dc5743be9c3547949 Mon Sep 17 00:00:00 2001
From: Bruce Martin <bruce@chanzuckerberg.com>
Date: Fri, 10 Apr 2020 13:52:02 -0600
Subject: [PATCH] fix non-finite condition in mean/var calculation (#1381)

---
 server/compute/diffexp.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/server/compute/diffexp.py b/server/compute/diffexp.py
index df476901..55ae9cbe 100644
--- a/server/compute/diffexp.py
+++ b/server/compute/diffexp.py
@@ -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