scikit-learn
diff --git a/‎sklearn/compose/tests/test_target.py
Copy file name to clipboardExpand all lines: sklearn/compose/tests/test_target.py
+3-4Lines changed: 3 additions & 4 deletions b/‎sklearn/compose/tests/test_target.py
Copy file name to clipboardExpand all lines: sklearn/compose/tests/test_target.py
+3-4Lines changed: 3 additions & 4 deletions
diff --git a/‎sklearn/neighbors/tests/test_nearest_centroid.py
Copy file name to clipboardExpand all lines: sklearn/neighbors/tests/test_nearest_centroid.py
-1Lines changed: 0 additions & 1 deletion b/‎sklearn/neighbors/tests/test_nearest_centroid.py
Copy file name to clipboardExpand all lines: sklearn/neighbors/tests/test_nearest_centroid.py
-1Lines changed: 0 additions & 1 deletion
@@ -8,7 +8,6 @@
 from sklearn.dummy import DummyRegressor
 
 from sklearn.utils._testing import assert_allclose
-from sklearn.utils._testing import assert_warns_message
 from sklearn.utils._testing import assert_no_warnings
 
 from sklearn.preprocessing import FunctionTransformer
@@ -54,9 +53,9 @@ def test_transform_target_regressor_invertible():
     regr = TransformedTargetRegressor(regressor=LinearRegression(),
                                       func=np.sqrt, inverse_func=np.log,
                                       check_inverse=True)
-    assert_warns_message(UserWarning, "The provided functions or transformer"
-                         " are not strictly inverse of each other.",
-                         regr.fit, X, y)
+    with pytest.warns(UserWarning, match="The provided functions or"
+                      " transformer are not strictly inverse of each other."):
+        regr.fit(X, y)
     regr = TransformedTargetRegressor(regressor=LinearRegression(),
                                       func=np.sqrt, inverse_func=np.log)
     regr.set_params(check_inverse=False)
 
@@ -1,7 +1,6 @@
 """
 Testing for the nearest centroid module.
 """
-
 import numpy as np
 import pytest
 from scipy import sparse as sp