From 6b5ecb50a9767d61555a824cf6999d3725a64834 Mon Sep 17 00:00:00 2001 From: Andreas Mueller Date: Wed, 2 Aug 2017 11:48:04 -0400 Subject: [PATCH] fix wrong assert in test_validation --- sklearn/model_selection/tests/test_validation.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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)