Ignore convergence warning in test_return_train_score_warn.

AlexandreSev · AlexandreSev · commit 3eb22a08b4ea · 2018-03-28T17:36:26.000+02:00
See scikit-learn#10866.
diff --git a/sklearn/model_selection/tests/test_search.py b/sklearn/model_selection/tests/test_search.py
@@ -33,6 +33,7 @@
 from sklearn.base import BaseEstimator
 from sklearn.base import clone
 from sklearn.exceptions import NotFittedError
+from sklearn.exceptions import ConvergenceWarning
 from sklearn.datasets import make_classification
 from sklearn.datasets import make_blobs
 from sklearn.datasets import make_multilabel_classification
@@ -344,11 +345,11 @@ def test_return_train_score_warn():
 
     estimators = [
         GridSearchCV(
-            LinearSVC(random_state=0, max_iter=100000),
+            LinearSVC(random_state=0),
             grid,
             iid=False),
         RandomizedSearchCV(
-            LinearSVC(random_state=0, max_iter=100000),
+            LinearSVC(random_state=0),
             grid,
             n_iter=2,
             iid=False)
@@ -358,7 +359,11 @@ def test_return_train_score_warn():
     for estimator in estimators:
         for val in [True, False, 'warn']:
             estimator.set_params(return_train_score=val)
-            result[val] = assert_no_warnings(estimator.fit, X, y).cv_results_
+            result[val] = assert_no_warnings(
+                ignore_warnings(estimator.fit, category=ConvergenceWarning),
+                X,
+                y
+                ).cv_results_
 
     train_keys = ['split0_train_score', 'split1_train_score',
                   'split2_train_score', 'mean_train_score', 'std_train_score']
