MNT use np.nonzero instead of np.where in _affinity_propagation.py (scikit-learn#30520)

emelia-hdz · jeremiedbb · web-flow · commit 7f325a933bae · 2025-04-11T19:38:23.000+02:00
Co-authored-by: Jérémie du Boisberranger &lt;jeremie@probabl.ai&gt;
diff --git a/sklearn/cluster/_affinity_propagation.py b/sklearn/cluster/_affinity_propagation.py
@@ -148,7 +148,7 @@ def _affinity_propagation(
         c[I] = np.arange(K)  # Identify clusters
         # Refine the final set of exemplars and clusters and return results
         for k in range(K):
-            ii = np.where(c == k)[0]
+            ii = np.asarray(c == k).nonzero()[0]
             j = np.argmax(np.sum(S[ii[:, np.newaxis], ii], axis=0))
             I[k] = ii[j]
 
