Add missing ... in weight_cdf[feature_idx, ...]

ogrisel · ogrisel · commit 020cbe833e49 · 2025-03-26T14:56:10.000+01:00
diff --git a/sklearn/utils/stats.py b/sklearn/utils/stats.py
@@ -94,7 +94,7 @@ def _weighted_percentile(array, sample_weight, percentile_rank=50):
     percentile_indices = xp.asarray(
         [
             xp.searchsorted(
-                weight_cdf[feature_idx], adjusted_percentile_rank[feature_idx]
+                weight_cdf[feature_idx, ...], adjusted_percentile_rank[feature_idx]
             )
             for feature_idx in range(weight_cdf.shape[0])
         ],

Original file line number	Diff line number	Diff line change
`@@ -94,7 +94,7 @@ def _weighted_percentile(array, sample_weight, percentile_rank=50):`
`94`	`94`	`percentile_indices = xp.asarray(`
`95`	`95`	`[`
`96`	`96`	`xp.searchsorted(`
`97`		`- weight_cdf[feature_idx], adjusted_percentile_rank[feature_idx]`
	`97`	`+ weight_cdf[feature_idx, ...], adjusted_percentile_rank[feature_idx]`
`98`	`98`	`)`
`99`	`99`	`for feature_idx in range(weight_cdf.shape[0])`
`100`	`100`	`],`