MAINT Parameters validation for sklearn.datasets.clear_data_home (#26259)

jiawei-zhang-a · jeremiedbb · web-flow · commit 18a45760092d · 2023-04-25T13:02:44.000Z
Co-authored-by: Jérémie du Boisberranger &lt;34657725+jeremiedbb@users.noreply.github.com&gt;
diff --git a/sklearn/datasets/_base.py b/sklearn/datasets/_base.py
@@ -68,12 +68,17 @@ def get_data_home(data_home=None) -> str:
     return data_home
 
 
+@validate_params(
+    {
+        "data_home": [str, os.PathLike, None],
+    }
+)
 def clear_data_home(data_home=None):
     """Delete all the content of the data home cache.
 
     Parameters
     ----------
-    data_home : str, default=None
+    data_home : str or path-like, default=None
         The path to scikit-learn data directory. If `None`, the default path
         is `~/sklearn_learn_data`.
     """
diff --git a/sklearn/tests/test_public_functions.py b/sklearn/tests/test_public_functions.py
@@ -119,6 +119,7 @@ def _check_function_param_validation(
     "sklearn.covariance.empirical_covariance",
     "sklearn.covariance.ledoit_wolf_shrinkage",
     "sklearn.covariance.shrunk_covariance",
+    "sklearn.datasets.clear_data_home",
     "sklearn.datasets.dump_svmlight_file",
     "sklearn.datasets.fetch_20newsgroups",
     "sklearn.datasets.fetch_20newsgroups_vectorized",
