optimized test (#890)

pyro-ppl · Jan 24, 2021 · f052ea6 · f052ea6
1 parent 2aabe8f
commit f052ea6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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])