File tree 2 files changed +4
-4
lines changed
2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change 33
33
BlockedStep ,
34
34
CompoundStep ,
35
35
StatsBijection ,
36
+ check_step_emits_tune ,
36
37
flat_statname ,
37
38
flatten_steps ,
38
39
)
@@ -207,11 +208,10 @@ def make_runmeta_and_point_fn(
207
208
) -> Tuple [mcb .RunMeta , PointFunc ]:
208
209
variables , point_fn = get_variables_and_point_fn (model , initial_point )
209
210
210
- sample_stats = [
211
- mcb .Variable ("tune" , "bool" ),
212
- ]
211
+ check_step_emits_tune (step )
213
212
214
213
# In PyMC the sampler stats are grouped by the sampler.
214
+ sample_stats = []
215
215
steps = flatten_steps (step )
216
216
for s , sm in enumerate (steps ):
217
217
for statname , (dtype , shape ) in sm .stats_dtypes_shapes .items ():
Original file line number Diff line number Diff line change @@ -119,7 +119,7 @@ def test_make_runmeta_and_point_fn(simple_model):
119
119
assert not vars ["vector" ].is_deterministic
120
120
assert not vars ["vector_interval__" ].is_deterministic
121
121
assert vars ["matrix" ].is_deterministic
122
- assert len (rmeta .sample_stats ) == 1 + len (step .stats_dtypes [0 ])
122
+ assert len (rmeta .sample_stats ) == len (step .stats_dtypes [0 ])
123
123
pass
124
124
125
125
You can’t perform that action at this time.
0 commit comments