MNT Add caller name to scale input validation (scikit-learn#31816)

Co-authored-by: L G Nithish Reddy <your-reddylnithish@gmail.com>
Co-authored-by: Jérémie du Boisberranger <jeremie@probabl.ai>

Author: 3 people

sklearn/preprocessing/_data.py

@@ -229,6 +229,7 @@ def scale(X, *, axis=0, with_mean=True, with_std=True, copy=True):
         estimator="the scale function",
         dtype=FLOAT_DTYPES,
         ensure_all_finite="allow-nan",
+        input_name="X",
     )
     if sparse.issparse(X):
         if with_mean:

sklearn/preprocessing/tests/test_data.py

@@ -1042,10 +1042,10 @@ def test_scale_sparse_with_mean_raise_exception(sparse_container):
 
 
 def test_scale_input_finiteness_validation():
-    # Check if non finite inputs raise ValueError
+    # Check if non-finite inputs raise ValueError
     X = [[np.inf, 5, 6, 7, 8]]
     with pytest.raises(
-        ValueError, match="Input contains infinity or a value too large"
+        ValueError, match=r"Input X contains infinity or a value too large for dtype"
     ):
         scale(X)