Reduce number of draws in slow BART test

pymc-devs · Jan 20, 2022 · 23c375a · 23c375a
1 parent da7c5df
commit 23c375a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pymc/tests/test_bart.py b/pymc/tests/test_bart.py
@@ -59,7 +59,7 @@ def test_missing_data():
         mu = pm.BART("mu", X, Y, m=10)
         sigma = pm.HalfNormal("sigma", 1)
         y = pm.Normal("y", mu, sigma, observed=Y)
-        idata = pm.sample(random_seed=3415)
+        idata = pm.sample(tune=10, draws=10, chains=1, random_seed=3415)
 
 
 class TestUtils: