Skip to content

Commit 6be1647

Browse files
author
Joseph Hall
committed
Fix failing tests
1 parent 4716abf commit 6be1647

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pymc/sampling/mcmc.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -138,11 +138,11 @@ def instantiate_steppers(
138138
if unused_args:
139139
s = "s" if len(unused_args) > 1 else ""
140140
example_arg = sorted(unused_args)[0]
141-
example_step = list(selected_steps.keys())[0]
141+
example_step = (list(selected_steps.keys()) or pm.STEP_METHODS)[0]
142142
raise ValueError(
143143
f"Invalid key{s} found in step_kwargs: {unused_args}. "
144144
"Keys must be step names and values valid kwargs for that stepper. "
145-
f'Did you mean {{"{example_step}": {{"{example_arg}": ...}}}}?'
145+
f'Did you mean {{"{example_step.name}": {{"{example_arg}": ...}}}}?'
146146
)
147147

148148
if len(steps) == 1:

0 commit comments

Comments
 (0)