MAINT Parameters validation for sklearn.preprocessing.add_dummy_feature (scikit-learn#26058)

ROMEEZHOU · web-flow · commit 1e942b2ec730 · 2023-04-03T12:42:39.000+02:00
diff --git a/sklearn/preprocessing/_data.py b/sklearn/preprocessing/_data.py
@@ -2337,6 +2337,12 @@ def _more_tags(self):
         return {"pairwise": True}
 
 
+@validate_params(
+    {
+        "X": ["array-like", "sparse matrix"],
+        "value": [Interval(Real, None, None, closed="neither")],
+    }
+)
 def add_dummy_feature(X, value=1.0):
     """Augment dataset with an additional dummy feature.
 
diff --git a/sklearn/tests/test_public_functions.py b/sklearn/tests/test_public_functions.py
@@ -214,6 +214,7 @@ def _check_function_param_validation(
     "sklearn.metrics.top_k_accuracy_score",
     "sklearn.metrics.zero_one_loss",
     "sklearn.model_selection.train_test_split",
+    "sklearn.preprocessing.add_dummy_feature",
     "sklearn.preprocessing.scale",
     "sklearn.random_projection.johnson_lindenstrauss_min_dim",
     "sklearn.svm.l1_min_c",
