@@ -226,7 +226,7 @@ class MyDocument(BaseDoc):
     metric_fn = getattr(comp_backend.Metrics, metric)
     dists = metric_fn(query_embeddings, index_embeddings, device=device)
     top_scores, top_indices = comp_backend.Retrieval.top_k(
-        dists, k=limit, device=device, descending=descending
+        dists, k=int(limit), device=device, descending=descending
     )
 
     batched_docs: List[DocList] = []
Original file line number	Diff line number	Diff line change
`@@ -226,7 +226,7 @@ class MyDocument(BaseDoc):`
`226`	`226`	`metric_fn = getattr(comp_backend.Metrics, metric)`
`227`	`227`	`dists = metric_fn(query_embeddings, index_embeddings, device=device)`
`228`	`228`	`top_scores, top_indices = comp_backend.Retrieval.top_k(`
`229`		`- dists, k=limit, device=device, descending=descending`
	`229`	`+ dists, k=int(limit), device=device, descending=descending`
`230`	`230`	`)`
`231`	`231`
`232`	`232`	`batched_docs: List[DocList] = []`