diff --git a/pynndescent/pynndescent_.py b/pynndescent/pynndescent_.py index 2aa73eb..aaaa4f9 100644 --- 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 index d591991..5e08ff4 100644 --- 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, )