d6fe384f6763c09da358e70375db034ec0e442f8,pynndescent/utils.py,,initalize_heap_from_graph_indices,#,764

Before Change

        for idx in range(graph_indices.shape[1]):
            j = graph_indices[i, idx]
            d = metric(data[i], data[j])
            unchecked_heap_push(heap, i, d, j, 1)

    return heap

After Change

            j = graph_indices[i, idx]
            d = metric(data[i], data[j])
            // unchecked_heap_push(heap, i, d, j, 1)
            flagged_heap_push(heap[0][i], heap[1][i], heap[2][i], j, d, 1)

    return heap

Instances