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_eval_metrics.py
+++ b/tests/python/test_eval_metrics.py
@@ -63,7 +63,7 @@ class TestEvalMetrics:
            from sklearn.cross_validation import train_test_split
        from sklearn.datasets import load_digits

-        digits = load_digits(2)
+        digits = load_digits(n_class=2)
        X = digits['data']
        y = digits['target']