diff --git a/python/scripts/index_faiss.py b/python/scripts/index_faiss.py index 6a57d7fe..6dd42392 100644 --- a/python/scripts/index_faiss.py +++ b/python/scripts/index_faiss.py @@ -4,7 +4,7 @@ import numpy as np from faiss import IndexHNSWFlat, IndexIVFPQ, read_index -from usearch.index import Matches +from usearch.index import Matches, BatchMatches from usearch.index import ( DEFAULT_CONNECTIVITY, DEFAULT_EXPANSION_ADD, @@ -45,9 +45,9 @@ def add(self, keys, vectors): # self._faiss.add_with_ids(vectors, keys) self._faiss.add(vectors) - def search(self, queries, k: int) -> Matches: + def search(self, queries, k: int) -> BatchMatches: distances, keys = self._faiss.search(queries, k) - return Matches(keys, distances, np.array([k] * queries.shape[0])) + return BatchMatches(keys, distances, np.array([k] * queries.shape[0])) def __len__(self) -> int: return self._faiss.ntotal diff --git a/python/usearch/index.py b/python/usearch/index.py index 23053577..2bbdb9b1 100644 --- a/python/usearch/index.py +++ b/python/usearch/index.py @@ -269,8 +269,8 @@ class Matches: keys: np.ndarray distances: np.ndarray - visited_members: int - computed_distances: int + visited_members: int = 0 + computed_distances: int = 0 def __len__(self) -> int: return len(self.keys) @@ -304,8 +304,8 @@ class BatchMatches: distances: np.ndarray counts: np.ndarray - visited_members: int - computed_distances: int + visited_members: int = 0 + computed_distances: int = 0 def __len__(self) -> int: return len(self.counts)