kind kwarg to argsort only supported in some numba versions

lmcinnes · Sep 7, 2019 · d855e61 · d855e61
1 parent 70a5c99
commit d855e61
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")
+        v = X[row].argsort()  # kind='quicksort' not supported by some numba versions
         v = v[:n_neighbors]
         knn_indices[row] = v
     return knn_indices