Skip to content

Commit 9d9ba38

Browse files
michaelraczyckimichaelosthege
authored andcommitted
implemented changes from PR review
1 parent f2bc9ba commit 9d9ba38

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

tests/sampling/test_mcmc.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -827,11 +827,12 @@ def test_step_vars_in_model(self):
827827
with pytensor.config.change_flags(mode=fast_unstable_sampling_mode):
828828
step1 = NUTS([c1])
829829
step2 = NUTS([c2])
830-
step2.vars = [
831-
at.dscalar("x"),
832-
]
830+
step2.vars = [c2]
833831
step = CompoundStep([step1, step2])
834-
with pytest.raises(ValueError):
832+
with pytest.raises(
833+
ValueError,
834+
match=r".* assigned to .* sampler is not a value variable in the model. You can use `util.get_value_vars_from_user_vars` to parse user provided variables.",
835+
):
835836
assign_step_methods(model, step)
836837

837838

0 commit comments

Comments
 (0)