diff --git a/sklearn/tests/test_base.py b/sklearn/tests/test_base.py index 740e83105c99173fb2e6f30e5820248c4109d58d..8112e7fd8196bb9b149efb159e9c2a03e2750b69 100644 --- a/sklearn/tests/test_base.py +++ b/sklearn/tests/test_base.py @@ -14,6 +14,7 @@ from sklearn.utils.testing import assert_raises from sklearn.utils.testing import assert_no_warnings from sklearn.utils.testing import assert_warns_message from sklearn.utils.testing import assert_dict_equal +from sklearn.utils.testing import ignore_warnings from sklearn.base import BaseEstimator, clone, is_classifier from sklearn.svm import SVC @@ -440,6 +441,7 @@ class SingleInheritanceEstimator(BaseEstimator): return data +@ignore_warnings(category=(UserWarning)) def test_pickling_works_when_getstate_is_overwritten_in_the_child_class(): estimator = SingleInheritanceEstimator() estimator._attribute_not_pickled = "this attribute should not be pickled"