diff --git a/sklearn/preprocessing/_data.py b/sklearn/preprocessing/_data.py index bbb6e6fa80a12..eb3d89b48dc5d 100644 --- a/sklearn/preprocessing/_data.py +++ b/sklearn/preprocessing/_data.py @@ -1651,6 +1651,9 @@ def _more_tags(self): return {"allow_nan": True} +@validate_params( + {"X": ["array-like", "sparse matrix"], "axis": [Options(Integral, {0, 1})]} +) def robust_scale( X, *, diff --git a/sklearn/tests/test_public_functions.py b/sklearn/tests/test_public_functions.py index a6c829b23620f..c7f6c4c2dea53 100644 --- a/sklearn/tests/test_public_functions.py +++ b/sklearn/tests/test_public_functions.py @@ -263,6 +263,7 @@ def test_function_param_validation(func_module): ("sklearn.decomposition.fastica", "sklearn.decomposition.FastICA"), ("sklearn.decomposition.non_negative_factorization", "sklearn.decomposition.NMF"), ("sklearn.preprocessing.minmax_scale", "sklearn.preprocessing.MinMaxScaler"), + ("sklearn.preprocessing.robust_scale", "sklearn.preprocessing.RobustScaler"), ]