HAVE_PYDESCENT to HAVE_DESCENT
@@ -90,7 +90,7 @@ class SlidingNNClustering:
assert HAVE_NUMBA, "SlidingNN needs numba to work"
assert HAVE_TORCH, "SlidingNN needs torch to work"
- assert HAVE_PYNNDESCENT, "SlidingNN needs pynndescent to work"
+ assert HAVE_NNDESCENT, "SlidingNN needs pynndescent to work"
assert HAVE_PYMDE, "SlidingNN needs pymde to work"
d = params
@@ -472,6 +472,7 @@ def build_windowed_nn_graph(
)
return nn_idx, nn_dist
+
if HAVE_NUMBA:
@numba.jit(fastmath=True, cache=True)
def sparse_euclidean(x, y, n_samples, n_dense):
