Skip to content

Commit d532c45

Browse files
committed
Deterministic make_shared_replacements
1 parent 5a83a9f commit d532c45

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

pymc/pytensorf.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -425,10 +425,11 @@ def make_shared_replacements(point, vars, model):
425425
-------
426426
Dict of variable -> new shared variable
427427
"""
428-
othervars = set(model.value_vars) - set(vars)
428+
vars_set = set(vars)
429429
return {
430430
var: pytensor.shared(point[var.name], var.name + "_shared", shape=var.type.shape)
431-
for var in othervars
431+
for var in model.value_vars
432+
if var not in vars_set
432433
}
433434

434435

0 commit comments

Comments
 (0)