Set n_neighbors as keyword arg for NearestNeighbors

KrishnaswamyLab · Aug 9, 2020 · be85023 · be85023
1 parent 712d5d0
commit be85023
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/graphtools/graphs.py b/graphtools/graphs.py
@@ -408,7 +408,9 @@ def build_kernel_to_data(
                     search_knn = min(search_knn * self.search_multiplier, knn_max)
                 if search_knn > self.data_nu.shape[0] / 2:
                     knn_tree = NearestNeighbors(
-                        search_knn, algorithm="brute", n_jobs=self.n_jobs
+                        n_neighbors=search_knn,
+                        algorithm="brute",
+                        n_jobs=self.n_jobs
                     ).fit(self.data_nu)
                 if len(update_idx) > 0:
                     if search_knn == knn_max: