Use isnull

Co-authored-by: Marco Gorelli <marcogorelli@protonmail.com>
pymc-devs · Jan 25, 2021 · cb57680 · cb57680
1 parent f0f8d83
commit cb57680
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pymc3/model.py b/pymc3/model.py
@@ -1698,7 +1698,7 @@ def pandas_to_array(data):
     if hasattr(data, "to_numpy"):
         # typically, but not limited to pandas objects
         vals = data.to_numpy()
-        mask = np.isnan(vals)
+        mask = data.isnull().to_numpy()
         if mask.any():
             # there are missing values
             ret = np.ma.MaskedArray(vals, mask)