From e059834c0fce4ad186433bec6a68fa2415f18bac Mon Sep 17 00:00:00 2001 From: Charlie-XIAO Date: Tue, 4 Apr 2023 06:21:01 +0800 Subject: [PATCH] MAINT Parameters validation for sklearn.metrics.pairwise.paired_euclidean_distances --- sklearn/metrics/pairwise.py | 7 +++++-- sklearn/tests/test_public_functions.py | 1 + 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/sklearn/metrics/pairwise.py b/sklearn/metrics/pairwise.py index 04ca260a041a4..3700eb0d8e1ff 100644 --- a/sklearn/metrics/pairwise.py +++ b/sklearn/metrics/pairwise.py @@ -1025,6 +1025,9 @@ def cosine_distances(X, Y=None): # Paired distances +@validate_params( + {"X": ["array-like", "sparse matrix"], "Y": ["array-like", "sparse matrix"]} +) def paired_euclidean_distances(X, Y): """Compute the paired euclidean distances between X and Y. @@ -1032,10 +1035,10 @@ def paired_euclidean_distances(X, Y): Parameters ---------- - X : array-like of shape (n_samples, n_features) + X : {array-like, sparse matrix} of shape (n_samples, n_features) Input array/matrix X. - Y : array-like of shape (n_samples, n_features) + Y : {array-like, sparse matrix} of shape (n_samples, n_features) Input array/matrix Y. Returns diff --git a/sklearn/tests/test_public_functions.py b/sklearn/tests/test_public_functions.py index b2d6b0da4f379..a0e85c6eec454 100644 --- a/sklearn/tests/test_public_functions.py +++ b/sklearn/tests/test_public_functions.py @@ -209,6 +209,7 @@ def _check_function_param_validation( "sklearn.metrics.pairwise.haversine_distances", "sklearn.metrics.pairwise.laplacian_kernel", "sklearn.metrics.pairwise.linear_kernel", + "sklearn.metrics.pairwise.paired_euclidean_distances", "sklearn.metrics.precision_recall_curve", "sklearn.metrics.precision_recall_fscore_support", "sklearn.metrics.precision_score",