File tree 2 files changed +5
-2
lines changed
2 files changed +5
-2
lines changed Original file line number Diff line number Diff line change @@ -89,7 +89,8 @@ def get_vars_in_point_list(trace, model):
89
89
names_in_trace = list (trace [0 ])
90
90
else :
91
91
names_in_trace = trace .varnames
92
- vars_in_trace = [model [v ] for v in names_in_trace if v in model ]
92
+ traceable_varnames = {var .name for var in (model .free_RVs + model .deterministics )}
93
+ vars_in_trace = [model [v ] for v in names_in_trace if v in traceable_varnames ]
93
94
return vars_in_trace
94
95
95
96
Original file line number Diff line number Diff line change @@ -1634,11 +1634,13 @@ def test_get_vars_in_point_list():
1634
1634
with pm .Model () as modelA :
1635
1635
pm .Normal ("a" , 0 , 1 )
1636
1636
pm .Normal ("b" , 0 , 1 )
1637
+ pm .Normal ("d" , 0 , 1 )
1637
1638
with pm .Model () as modelB :
1638
1639
a = pm .Normal ("a" , 0 , 1 )
1639
1640
pm .Normal ("c" , 0 , 1 )
1641
+ pm .ConstantData ("d" , 0 )
1640
1642
1641
- point_list = [{"a" : 0 , "b" : 0 }]
1643
+ point_list = [{"a" : 0 , "b" : 0 , "d" : 0 }]
1642
1644
vars_in_trace = get_vars_in_point_list (point_list , modelB )
1643
1645
assert set (vars_in_trace ) == {a }
1644
1646
You can’t perform that action at this time.
0 commit comments