Fix flaky test_change_point_x64

pyro-ppl · Sep 21, 2024 · 9535274 · 9535274
1 parent 3cde93d
commit 9535274
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/infer/test_mcmc.py b/test/infer/test_mcmc.py
@@ -346,7 +346,7 @@ def model():
 
 def test_change_point_x64():
     # Ref: https://forum.pyro.ai/t/i-dont-understand-why-nuts-code-is-not-working-bayesian-hackers-mail/696
-    num_warmup, num_samples = 500, 3000
+    num_warmup, num_samples = 1000, 3000
 
     def model(data):
         alpha = 1 / jnp.mean(data.astype(np.float32))