Skip to content

Commit 5d8dd32

Browse files
Improve isinstance statement
Co-authored-by: Osvaldo A Martin <[email protected]>
1 parent 4406a15 commit 5d8dd32

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pymc_bart/bart.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ def rng_fn( # pylint: disable=W0237
5555
if not size:
5656
size = None
5757

58-
if isinstance(cls.Y, TensorSharedVariable) or isinstance(cls.Y, TensorVariable):
58+
if isinstance(cls.Y, (TensorSharedVariable, TensorVariable)):
5959
Y = cls.Y.eval()
6060
else:
6161
Y = cls.Y

0 commit comments

Comments
 (0)