diff --git a/test/test_mcmc.py b/test/test_mcmc.py index 0e8aaf730..1a667337c 100644 --- a/test/test_mcmc.py +++ b/test/test_mcmc.py @@ -484,7 +484,7 @@ def model(): @pytest.mark.parametrize('algo', ['HMC', 'NUTS']) def test_functional_beta_bernoulli_x64(algo): - warmup_steps, num_samples = 500, 20000 + warmup_steps, num_samples = 410, 100 def model(data): alpha = jnp.array([1.1, 1.1])