Fix issue with fast_knn_indices

lmcinnes · Sep 7, 2019 · 70bc1e5 · 70bc1e5
1 parent d855e61
commit 70bc1e5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/umap/utils.py b/umap/utils.py
@@ -27,7 +27,7 @@ def fast_knn_indices(X, n_neighbors):
     """
     knn_indices = np.empty((X.shape[0], n_neighbors), dtype=np.int32)
     for row in numba.prange(X.shape[0]):
-        v = X[row].argsort()  # kind='quicksort' not supported by some numba versions
+        v = np.argsort(X[row])  # Need to call argsort this way for numba
         v = v[:n_neighbors]
         knn_indices[row] = v
     return knn_indices