From 26a82621a2a3e34a2ee997780617f255e356a580 Mon Sep 17 00:00:00 2001 From: Vadim Khotilovich Date: Sat, 4 Jun 2016 22:38:24 -0500 Subject: [PATCH] make travis happy --- tests/python/test_basic_models.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/python/test_basic_models.py b/tests/python/test_basic_models.py index 1da738e89..9e9c08423 100644 --- a/tests/python/test_basic_models.py +++ b/tests/python/test_basic_models.py @@ -111,6 +111,7 @@ class TestModels(unittest.TestCase): param['eval_metric'] = ["auc", "logloss", 'error'] evals_result = {} bst = xgb.train(param, dtrain, 4, watchlist, evals_result=evals_result) + assert isinstance(bst, xgb.core.Booster) assert len(evals_result['eval']) == 3 assert set(evals_result['eval'].keys()) == {'auc', 'error', 'logloss'}