diff --git a/sklearn/model_selection/tests/test_validation.py b/sklearn/model_selection/tests/test_validation.py index dedb77026c544..5f650cb644079 100644 --- a/sklearn/model_selection/tests/test_validation.py +++ b/sklearn/model_selection/tests/test_validation.py @@ -452,8 +452,8 @@ def check_cross_validate_multi_metric(clf, X, y, scores): assert type(cv_results['test_r2']) == np.ndarray assert (type(cv_results['test_neg_mean_squared_error']) == np.ndarray) - assert type(cv_results['fit_time'] == np.ndarray) - assert type(cv_results['score_time'] == np.ndarray) + assert type(cv_results['fit_time']) == np.ndarray + assert type(cv_results['score_time']) == np.ndarray # Ensure all the times are within sane limits assert np.all(cv_results['fit_time'] >= 0)