Skip to content

Commit da836dd

Browse files
committed
fix flaky vb unit test
1 parent 90e7bf8 commit da836dd

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

test/test_variational.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -242,7 +242,11 @@ def test_exe_only(self):
242242

243243
bern2_model = CmdStanModel(exe_file=exe_only)
244244
jdata = os.path.join(DATAFILES_PATH, 'bernoulli.data.json')
245-
variational = bern2_model.variational(data=jdata, algorithm='meanfield')
245+
variational = bern2_model.variational(
246+
data=jdata,
247+
require_converged=False,
248+
seed=12345,
249+
algorithm='meanfield')
246250
self.assertEqual(variational.variational_sample.shape, (1000, 4))
247251

248252

0 commit comments

Comments
 (0)