Skip to content

Commit e552245

Browse files
committed
Deterministic make_shared_replacements
1 parent 6993a2f commit e552245

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
@@ -412,10 +412,11 @@ def make_shared_replacements(point, vars, model):
412412
-------
413413
Dict of variable -> new shared variable
414414
"""
415-
othervars = set(model.value_vars) - set(vars)
415+
vars_set = set(vars)
416416
return {
417417
var: pytensor.shared(point[var.name], var.name + "_shared", shape=var.type.shape)
418-
for var in othervars
418+
for var in model.value_vars
419+
if var not in vars_set
419420
}
420421

421422

0 commit comments

Comments
 (0)