fixup! DOC Better document implementation

jjerphan · jjerphan · commit 6a3f58ced5e4 · 2023-03-09T10:30:55.000+01:00
diff --git a/sklearn/metrics/_pairwise_distances_reduction/_middle_term_computer.pyx.tp b/sklearn/metrics/_pairwise_distances_reduction/_middle_term_computer.pyx.tp
@@ -49,9 +49,9 @@ cdef void _middle_term_sparse_sparse_64(
     const SPARSE_INDEX_TYPE_t[:] Y_indptr,
     ITYPE_t Y_start,
     ITYPE_t Y_end,
-    DTYPE_t * dist_middle_terms,
+    DTYPE_t * D,
 ) noexcept nogil:
-    # This routine assumes that D is a pointer to the first element of a
+    # This routine assumes that D points to the first element of a
     # zeroed buffer of length at least equal to n_X × n_Y, conceptually
     # representing a 2-d C-ordered array.
     cdef:
@@ -68,7 +68,7 @@ cdef void _middle_term_sparse_sparse_64(
                 for y_ptr in range(Y_indptr[Y_start+j], Y_indptr[Y_start+j+1]):
                     y_col = Y_indices[y_ptr]
                     if x_col == y_col:
-                        dist_middle_terms[k] += -2 * X_data[x_ptr] * Y_data[y_ptr]
+                        D[k] += -2 * X_data[x_ptr] * Y_data[y_ptr]
 
 
 {{for name_suffix, upcast_to_float64, INPUT_DTYPE_t, INPUT_DTYPE in implementation_specific_values}}
