Mark pm.Data tensors.

pymc-devs · May 23, 2019 · 29c72ec · 29c72ec
1 parent 6a6da63
commit 29c72ec
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pymc3/data.py b/pymc3/data.py
@@ -390,7 +390,7 @@ def align_minibatches(batches=None):
 
 class Data:
     """Data container class that wraps the theano SharedVariable class
-    and let the model be aware of its inputs and outputs.
+    and lets the model be aware of its inputs and outputs.
 
     Parameters
     ----------
@@ -449,4 +449,7 @@ def __new__(self, name, value):
                             "inside a 'with model:' block.")
         model.add_random_variable(shared_object)
 
+        # mark this as a data object
+        shared_object.is_data = True
+
         return shared_object