MAINT Parameters validation for sklearn.metrics.calinski_harabasz_score (#26135)

jiawei-zhang-a · web-flow · commit e74041fd9d4f · 2023-04-12T15:03:36.000+02:00
diff --git a/sklearn/metrics/cluster/_unsupervised.py b/sklearn/metrics/cluster/_unsupervised.py
@@ -298,6 +298,12 @@ def silhouette_samples(X, labels, *, metric="euclidean", **kwds):
     return np.nan_to_num(sil_samples)
 
 
+@validate_params(
+    {
+        "X": ["array-like"],
+        "labels": ["array-like"],
+    }
+)
 def calinski_harabasz_score(X, labels):
     """Compute the Calinski and Harabasz score.
 
diff --git a/sklearn/tests/test_public_functions.py b/sklearn/tests/test_public_functions.py
@@ -167,6 +167,7 @@ def _check_function_param_validation(
     "sklearn.metrics.average_precision_score",
     "sklearn.metrics.balanced_accuracy_score",
     "sklearn.metrics.brier_score_loss",
+    "sklearn.metrics.calinski_harabasz_score",
     "sklearn.metrics.class_likelihood_ratios",
     "sklearn.metrics.classification_report",
     "sklearn.metrics.cluster.adjusted_mutual_info_score",
