bug in arguments to batch search

cmuparlay · Nov 13, 2023 · 7d01ab9 · 7d01ab9
1 parent c0149ec
commit 7d01ab9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/graph_index.cpp b/python/graph_index.cpp
@@ -53,7 +53,7 @@ struct GraphIndex{
     }
 
     NeighborsAndDistances batch_search(py::array_t<T, py::array::c_style | py::array::forcecast> &queries, uint64_t num_queries, uint64_t knn,
-                        uint64_t beam_width, uint64_t visit_limit = -1){
+                        uint64_t beam_width, int64_t visit_limit = -1){
         if(visit_limit == -1) visit_limit = G.size();
         QueryParams QP(knn, beam_width, 1.35, visit_limit, G.max_degree());