fix wrong assert in test_validation (#9480)

amueller · jnothman · commit 0dd19de33a79 · 2017-08-03T07:39:46.000+10:00
diff --git 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)
