diff --git a/vicinity/backends/faiss.py b/vicinity/backends/faiss.py index f438e0d..74aff17 100644 --- a/vicinity/backends/faiss.py +++ b/vicinity/backends/faiss.py @@ -190,7 +190,7 @@ def threshold(self, vectors: npt.NDArray, threshold: float) -> list[npt.NDArray] out.append(within_threshold_indices) else: # Fallback to search-based filtering for indexes that do not support range_search - distances, indices = self.index.search(vectors, 100) # Arbitrarily large `k` to capture potential matches + distances, indices = self.index.search(vectors, 100) for dist, idx in zip(distances, indices): # Convert distances for cosine if needed