Skip to content

Commit 78b1380

Browse files
Larry Dongtwiecki
authored andcommitted
Naive fix to pm.sample_prior_predictive take 2: excluding TensorConstants in inputs to compile_rv_inplace, fixed typo
1 parent 9dad9c2 commit 78b1380

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

pymc3/sampling.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@
3131
import xarray
3232

3333
from aesara.compile.mode import Mode
34+
from aesara.graph.basic import Constant
3435
from aesara.tensor.sharedvar import SharedVariable
3536
from arviz import InferenceData
3637
from fastprogress.fastprogress import progress_bar
@@ -2001,7 +2002,7 @@ def sample_prior_predictive(
20012002
names.append(rv_var.name)
20022003
vars_to_sample.append(rv_var)
20032004

2004-
inputs = [i for i in inputvars(vars_to_sample) if not isinstance(i, SharedVariable)]
2005+
inputs = [i for i in inputvars(vars_to_sample) if not isinstance(i, (SharedVariable, Constant))]
20052006

20062007
sampler_fn = compile_rv_inplace(
20072008
inputs, vars_to_sample, allow_input_downcast=True, accept_inplace=True, mode=mode

0 commit comments

Comments
 (0)