d6fe384f6763c09da358e70375db034ec0e442f8,pynndescent/sparse_nndescent.py,,init_rp_tree,#,267

Before Change

                if p == -1 or q == -1:
                    continue

                heap_push(current_graph, p, d, q, 1)
                heap_push(current_graph, q, d, p, 1)


@numba.njit(fastmath=True)
def init_random(n_neighbors, inds, indptr, data, heap, dist, rng_state):

After Change

                // heap_push(current_graph, p, d, q, 1)
                // heap_push(current_graph, q, d, p, 1)
                checked_flagged_heap_push(
                    current_graph[1][p],
                    current_graph[0][p],
                    current_graph[2][p],
                    d,
                    q,
                    np.uint8(1),
                )
                checked_flagged_heap_push(
                    current_graph[1][q],
                    current_graph[0][q],
                    current_graph[2][q],
                    d,
                    p,
                    np.uint8(1),
                )


@numba.njit(fastmath=True, locals={
    "d": numba.float32,

Instances