fix: pass limit as integer (#1657)

Signed-off-by: Joan Fontanals Martinez <joan.martinez@jina.ai>
docarray · Jun 16, 2023 · dc96e38 · dc96e38
1 parent 7e211a9
commit dc96e38
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docarray/index/backends/hnswlib.py b/docarray/index/backends/hnswlib.py
@@ -297,7 +297,7 @@ def _find_batched(
         limit = min(limit, self.num_docs())
 
         index = self._hnsw_indices[search_field]
-        labels, distances = index.knn_query(queries, k=limit)
+        labels, distances = index.knn_query(queries, k=int(limit))
         result_das = [
             self._get_docs_sqlite_hashed_id(
                 ids_per_query.tolist(),