Skip to content

Commit 7837629

Browse files
committed
fix test
1 parent 3f94827 commit 7837629

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/test_bart.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
from numpy.testing import assert_almost_equal, assert_array_equal
66
from pymc.tests.distributions.util import assert_moment_is_expected
77

8+
import pymc_bart as pmb
89

910
def test_split_node():
1011
split_node = pmb.tree.SplitNode(index=5, idx_split_variable=2, split_value=3.0)

0 commit comments

Comments
 (0)