Make heap doesn't need a signature and this might make numba happier
@@ -100,7 +100,7 @@ def rejection_sample(n_samples, pool_size, rng_state):
return result
-@numba.njit('f8[:, :, :](i8,i8)')
+@numba.njit()
def make_heap(n_points, size):
"""Constructor for the numba enabled heap objects. The heaps are used
for approximate nearest neighbor search, maintaining a list of potential
