Skip to content

Commit 6a299a3

Browse files
authored
Added random_seed argument to the sampler in some MLDA tests (#4313)
1 parent a9806db commit 6a299a3

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

pymc3/tests/test_step.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -1133,7 +1133,7 @@ def test_nonparallelized_chains_are_random(self):
11331133
Normal("x", 0, 1)
11341134
for stepper in TestMLDA.steppers:
11351135
step = stepper(coarse_models=[coarse_model])
1136-
trace = sample(chains=2, cores=1, draws=20, tune=0, step=step)
1136+
trace = sample(chains=2, cores=1, draws=20, tune=0, step=step, random_seed=1)
11371137
samples = np.array(trace.get_values("x", combine=False))[:, 5]
11381138
assert (
11391139
len(set(samples)) == 2
@@ -1150,7 +1150,7 @@ def test_parallelized_chains_are_random(self):
11501150
Normal("x", 0, 1)
11511151
for stepper in TestMLDA.steppers:
11521152
step = stepper(coarse_models=[coarse_model])
1153-
trace = sample(chains=2, cores=2, draws=20, tune=0, step=step)
1153+
trace = sample(chains=2, cores=2, draws=20, tune=0, step=step, random_seed=1)
11541154
samples = np.array(trace.get_values("x", combine=False))[:, 5]
11551155
assert len(set(samples)) == 2, "Parallelized {} " "chains are identical.".format(
11561156
stepper
@@ -1176,7 +1176,7 @@ def test_acceptance_rate_against_coarseness(self):
11761176
Normal("x", 5.0, 1.0)
11771177
for coarse_model in possible_coarse_models:
11781178
step = MLDA(coarse_models=[coarse_model], subsampling_rates=3)
1179-
trace = sample(chains=1, draws=500, tune=100, step=step)
1179+
trace = sample(chains=1, draws=500, tune=100, step=step, random_seed=1)
11801180
acc.append(trace.get_sampler_stats("accepted").mean())
11811181
assert acc[0] > acc[1] > acc[2], (
11821182
"Acceptance rate is not "

0 commit comments

Comments
 (0)