From f72f5be77259b07bac44e214d916e3fd43c1ee4f Mon Sep 17 00:00:00 2001 From: scibol Date: Fri, 20 Dec 2019 15:17:53 +0100 Subject: [PATCH] fixed integer test for train and test indices --- sklearn/model_selection/tests/test_split.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sklearn/model_selection/tests/test_split.py b/sklearn/model_selection/tests/test_split.py index eb5eb192a6921..253593968ad24 100644 --- a/sklearn/model_selection/tests/test_split.py +++ b/sklearn/model_selection/tests/test_split.py @@ -175,7 +175,7 @@ def test_cross_validator_with_default_params(): # Test that train, test indices returned are integers for train, test in cv.split(X, y, groups): assert np.asarray(train).dtype.kind == 'i' - assert np.asarray(train).dtype.kind == 'i' + assert np.asarray(test).dtype.kind == 'i' # Test if the repr works without any errors assert cv_repr == repr(cv)