Support imputations with ndarray data

closes #4437
pymc-devs · Jan 24, 2021 · c183d11 · c183d11
1 parent 823906a
commit c183d11
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/RELEASE-NOTES.md b/RELEASE-NOTES.md
@@ -5,6 +5,7 @@
 ### Breaking Changes
 
 ### New Features
++ Automatic imputations now also works with `ndarray` data (see[#4439](https://github.com/pymc-devs/pymc3/pull/4439)).
 
 ### Maintenance
 - `math.log1mexp_numpy` no longer raises RuntimeWarning when given very small inputs. These were commonly observed during NUTS sampling (see [#4428](https://github.com/pymc-devs/pymc3/pull/4428)).

diff --git a/pymc3/model.py b/pymc3/model.py
@@ -1695,7 +1695,13 @@ def pandas_to_array(data):
     XXX: When `data` is a generator, this will return a Theano tensor!
 
     """
-    if hasattr(data, "values"):  # pandas
+    if isinstance(data, np.ndarray):
+        mask = np.isnan(data)
+        if np.any(mask):
+            ret = np.ma.MaskedArray(data, mask)
+        else:
+            ret = data
+    elif hasattr(data, "values"):  # pandas
         if data.isnull().any().any():  # missing values
             ret = np.ma.MaskedArray(data.values, data.isnull().values)
         else: