Fix for python 3

2015-08-23 13:37:29 +09:00
parent 8e06726f6b
commit 00702dc39b
5 changed files with 33 additions and 11 deletions
--- a/tests/python/test_basic.py
+++ b/tests/python/test_basic.py
@@ -33,6 +33,9 @@ def test_plotting():
    bst2 = xgb.Booster(model_file='xgb.model')
    # plotting

+    import matplotlib
+    matplotlib.use('Agg')
+
    from matplotlib.axes import Axes
    from graphviz import Digraph