diff --git a/sklearn/tests/test_common.py b/sklearn/tests/test_common.py index 730f1135b833a..33c9b712a873f 100644 --- a/sklearn/tests/test_common.py +++ b/sklearn/tests/test_common.py @@ -19,6 +19,7 @@ from sklearn.utils import all_estimators from sklearn.utils._testing import ignore_warnings from sklearn.exceptions import ConvergenceWarning +from sklearn.exceptions import FitFailedWarning from sklearn.utils.estimator_checks import check_estimator import sklearn @@ -246,6 +247,7 @@ def test_search_cv(estimator, check, request): ConvergenceWarning, UserWarning, FutureWarning, + FitFailedWarning, ) ): check(estimator)