Keyword arg for the sparse case

lmcinnes · Sep 11, 2019 · ac0b43d · ac0b43d
1 parent 494024f
commit ac0b43d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/umap/umap_.py b/umap/umap_.py
@@ -326,7 +326,7 @@ def nearest_neighbors(
                     n_neighbors,
                     rng_state,
                     max_candidates=60,
-                    dist=distance_func,
+                    sparse_dist=distance_func,
                     dist_args=tuple(metric_kwds.values()),
                     low_memory=low_memory,
                     rp_tree_init=True,