Fix as per jlmellville

lmcinnes · Sep 9, 2019 · 679527c · 679527c
1 parent bc6106c
commit 679527c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/pynndescent/pynndescent_.py b/pynndescent/pynndescent_.py
@@ -223,13 +223,13 @@ def nn_descent_internal_high_memory(
     data,
     n_neighbors,
     rng_state,
+    tried,
     max_candidates=50,
     dist=dist.euclidean,
     dist_args=(),
     n_iters=10,
     delta=0.001,
     rho=0.5,
-    tried=set([(-1, -1)]),
     verbose=False,
     seed_per_row=False,
 ):
@@ -338,13 +338,13 @@ def nn_descent(
             data,
             n_neighbors,
             rng_state,
+            tried,
             max_candidates=max_candidates,
             dist=dist,
             dist_args=dist_args,
             n_iters=n_iters,
             delta=delta,
             rho=rho,
-            tried=tried,
             verbose=verbose,
             seed_per_row=seed_per_row,
         )

diff --git a/pynndescent/sparse_nndescent.py b/pynndescent/sparse_nndescent.py
@@ -140,13 +140,13 @@ def sparse_nn_descent_internal_high_memory(
     n_vertices,
     n_neighbors,
     rng_state,
+    tried,
     max_candidates=50,
     sparse_dist=sparse_euclidean,
     dist_args=(),
     n_iters=10,
     delta=0.001,
     rho=0.5,
-    tried=set([(-1, -1)]),
     verbose=False,
 ):
     for n in range(n_iters):
@@ -289,13 +289,13 @@ def sparse_nn_descent(
             n_vertices,
             n_neighbors,
             rng_state,
+            tried,
             max_candidates=max_candidates,
             sparse_dist=sparse_dist,
             dist_args=dist_args,
             n_iters=n_iters,
             delta=delta,
             rho=rho,
-            tried=tried,
             verbose=verbose,
         )