MAINT Parameters validation for sklearn.preprocessing.robust_scale (#26086)

2357juan · web-flow · commit 24f68c2a9e6a · 2023-04-12T12:15:23.000+02:00
diff --git 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
@@ -269,6 +269,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"),
 ]
 
 

Original file line number	Diff line number	Diff line change
`@@ -269,6 +269,7 @@ def test_function_param_validation(func_module):`
`269`	`269`	`("sklearn.decomposition.fastica", "sklearn.decomposition.FastICA"),`
`270`	`270`	`("sklearn.decomposition.non_negative_factorization", "sklearn.decomposition.NMF"),`
`271`	`271`	`("sklearn.preprocessing.minmax_scale", "sklearn.preprocessing.MinMaxScaler"),`
	`272`	`+ ("sklearn.preprocessing.robust_scale", "sklearn.preprocessing.RobustScaler"),`
`272`	`273`	`]`
`273`	`274`
`274`	`275`