Remove warnings in tests. (#6554)

2020-12-31 13:41:18 +08:00
parent 8ad22bf4e7
commit 5e9e525223
3 changed files with 8 additions and 5 deletions
--- a/tests/python/test_early_stopping.py
+++ b/tests/python/test_early_stopping.py
@@ -16,7 +16,7 @@ class TestEarlyStopping:
        except ImportError:
            from sklearn.cross_validation import train_test_split

-        digits = load_digits(2)
+        digits = load_digits(n_class=2)
        X = digits['data']
        y = digits['target']
        X_train, X_test, y_train, y_test = train_test_split(X, y,
@@ -52,7 +52,7 @@ class TestEarlyStopping:
    def test_cv_early_stopping(self):
        from sklearn.datasets import load_digits

-        digits = load_digits(2)
+        digits = load_digits(n_class=2)
        X = digits['data']
        y = digits['target']
        dm = xgb.DMatrix(X, label=y)