From ea4d82c79a4b564cad356b8ffab01859c40ce870 Mon Sep 17 00:00:00 2001 From: Charlie-XIAO Date: Thu, 13 Apr 2023 00:43:35 +0800 Subject: [PATCH] MAINT Parameters validation for sklearn.datasets.fetch_species_distributions --- sklearn/datasets/_species_distributions.py | 2 ++ sklearn/tests/test_public_functions.py | 1 + 2 files changed, 3 insertions(+) diff --git a/sklearn/datasets/_species_distributions.py b/sklearn/datasets/_species_distributions.py index 091708e6f8ee4..2aa7caab403c8 100644 --- a/sklearn/datasets/_species_distributions.py +++ b/sklearn/datasets/_species_distributions.py @@ -51,6 +51,7 @@ from ._base import RemoteFileMetadata from ..utils import Bunch from ._base import _pkl_filepath +from ..utils._param_validation import validate_params # The original data can be found at: # https://biodiversityinformatics.amnh.org/open_source/maxent/samples.zip @@ -137,6 +138,7 @@ def construct_grids(batch): return (xgrid, ygrid) +@validate_params({"data_home": [str, None], "download_if_missing": ["boolean"]}) def fetch_species_distributions(*, data_home=None, download_if_missing=True): """Loader for species distribution dataset from Phillips et. al. (2006). diff --git a/sklearn/tests/test_public_functions.py b/sklearn/tests/test_public_functions.py index 62513b0f63cce..bd09921346df3 100644 --- a/sklearn/tests/test_public_functions.py +++ b/sklearn/tests/test_public_functions.py @@ -128,6 +128,7 @@ def _check_function_param_validation( "sklearn.datasets.fetch_lfw_people", "sklearn.datasets.fetch_olivetti_faces", "sklearn.datasets.fetch_rcv1", + "sklearn.datasets.fetch_species_distributions", "sklearn.datasets.load_svmlight_file", "sklearn.datasets.load_svmlight_files", "sklearn.datasets.make_biclusters",