From be95c80aa29233f7089f54befaa8727607c7d9b4 Mon Sep 17 00:00:00 2001 From: Joosep Date: Tue, 14 Jul 2015 11:38:38 +0200 Subject: [PATCH] fix wrapper dict --- wrapper/xgboost.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/wrapper/xgboost.py b/wrapper/xgboost.py index 96f6c2573..7a601424c 100644 --- a/wrapper/xgboost.py +++ b/wrapper/xgboost.py @@ -777,7 +777,7 @@ def train(params, dtrain, num_boost_round=10, evals=(), obj=None, feval=None, bst = Booster(params, [dtrain] + [d[0] for d in evals]) if evals_result is not None: - if isinstance(evals_result, dict): + if not isinstance(evals_result, dict): raise TypeError('evals_result has to be a dictionary') else: evals_name = [d[1] for d in evals]