fix wrong assert in test_validation (scikit-learn#9480)

maskani-moh · Nov 15, 2017 · a06fad2 · a06fad2
1 parent c20862d
commit a06fad2
Showing 1 changed file with 2 additions and 2 deletions.
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)