From 05d89defddaccb8d61407831c574683082de600c Mon Sep 17 00:00:00 2001 From: Lukas Geiger Date: Fri, 15 Dec 2023 13:23:32 +0000 Subject: [PATCH] MAINT: Prefer `np.fill_diagonal` over `diag_indices` --- sklearn/metrics/pairwise.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sklearn/metrics/pairwise.py b/sklearn/metrics/pairwise.py index 5d848dae5b11f..4e5c37dff0091 100644 --- a/sklearn/metrics/pairwise.py +++ b/sklearn/metrics/pairwise.py @@ -1084,7 +1084,7 @@ def cosine_distances(X, Y=None): if X is Y or Y is None: # Ensure that distances between vectors and themselves are set to 0.0. # This may not be the case due to floating point rounding errors. - S[np.diag_indices_from(S)] = 0.0 + np.fill_diagonal(S, 0.0) return S