Skip to content

Commit 23c375a

Browse files
ricardoV94twiecki
authored andcommitted
Reduce number of draws in slow BART test
1 parent da7c5df commit 23c375a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pymc/tests/test_bart.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ def test_missing_data():
5959
mu = pm.BART("mu", X, Y, m=10)
6060
sigma = pm.HalfNormal("sigma", 1)
6161
y = pm.Normal("y", mu, sigma, observed=Y)
62-
idata = pm.sample(random_seed=3415)
62+
idata = pm.sample(tune=10, draws=10, chains=1, random_seed=3415)
6363

6464

6565
class TestUtils:

0 commit comments

Comments
 (0)