We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
isinstance
1 parent 4406a15 commit 5d8dd32Copy full SHA for 5d8dd32
pymc_bart/bart.py
@@ -55,7 +55,7 @@ def rng_fn( # pylint: disable=W0237
55
if not size:
56
size = None
57
58
- if isinstance(cls.Y, TensorSharedVariable) or isinstance(cls.Y, TensorVariable):
+ if isinstance(cls.Y, (TensorSharedVariable, TensorVariable)):
59
Y = cls.Y.eval()
60
else:
61
Y = cls.Y
0 commit comments