Skip to content

Commit 5a7b552

Browse files
committed
fix test
1 parent a473028 commit 5a7b552

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pymc3/tests/test_sampling.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -166,7 +166,7 @@ def test_trace_report(self, step_cls, discard):
166166
assert isinstance(trace.report.t_sampling, float)
167167
pass
168168

169-
def test_trace_report_bart():
169+
def test_trace_report_bart(self):
170170
X = np.random.normal(0, 1, size=(3, 250)).T
171171
Y = np.random.normal(0, 1, size=250)
172172
X[:, 0] = np.random.normal(Y, 0.1)
@@ -175,7 +175,7 @@ def test_trace_report_bart():
175175
mu = pm.BART("mu", X, Y, m=20)
176176
sigma = pm.HalfNormal("sigma", 1)
177177
y = pm.Normal("y", mu, sigma, observed=Y)
178-
trace = pm.sample(500, tune=100, chains=1, random_seed=3415)
178+
trace = pm.sample(500, tune=100, random_seed=3415)
179179
var_imp = trace.report.variable_importance
180180
assert var_imp[0] > var_imp[1:].sum()
181181
npt.assert_almost_equal(var_imp.sum(), 1)

0 commit comments

Comments
 (0)