disable progress bar for callable chain_method tests

pyro-ppl · Dec 4, 2024 · 83d6efe · 83d6efe
1 parent 602cd1b
commit 83d6efe
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/test/infer/test_hmc_gibbs.py b/test/infer/test_hmc_gibbs.py
@@ -478,7 +478,12 @@ def gibbs_fn(rng_key, gibbs_sites, hmc_sites):
     hmc_kernel = NUTS(model)
     kernel = HMCGibbs(hmc_kernel, gibbs_fn=gibbs_fn, gibbs_sites=["x"])
     mcmc = MCMC(
-        kernel, num_warmup=100, num_chains=2, num_samples=100, chain_method=vmap
+        kernel,
+        num_warmup=100,
+        num_chains=2,
+        num_samples=100,
+        chain_method=vmap,
+        progress_bar=False,
     )
     mcmc.run(random.PRNGKey(0))
     samples = mcmc.get_samples()