Patch for case when Y is a TensorVariable (#206)

AlexAndorra · aloctavodia · web-flow · commit 77116d1289c4 · 2024-12-19T15:02:13.000-03:00
* add case tensor var for Y

* Improve `isinstance` statement

Co-authored-by: Osvaldo A Martin &lt;aloctavodia@gmail.com&gt;

---------

Co-authored-by: Osvaldo A Martin &lt;aloctavodia@gmail.com&gt;
diff --git a/pymc_bart/bart.py b/pymc_bart/bart.py
@@ -26,6 +26,7 @@
 from pymc.logprob.abstract import _logprob
 from pytensor.tensor.random.op import RandomVariable
 from pytensor.tensor.sharedvar import TensorSharedVariable
+from pytensor.tensor.variable import TensorVariable
 
 from .split_rules import SplitRule
 from .tree import Tree
@@ -54,7 +55,7 @@ def rng_fn(  # pylint: disable=W0237
         if not size:
             size = None
 
-        if isinstance(cls.Y, TensorSharedVariable):
+        if isinstance(cls.Y, (TensorSharedVariable, TensorVariable)):
             Y = cls.Y.eval()
         else:
             Y = cls.Y
