Skip to content

Commit

Permalink
minor refactoring
Browse files Browse the repository at this point in the history
  • Loading branch information
pmpailis committed Nov 24, 2023
1 parent 134baa1 commit faf17fd
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions dense_vector/track.py
Original file line number Diff line number Diff line change
Expand Up @@ -155,9 +155,8 @@ def __init__(self, track, params, **kwargs):
self.infinite = True
self._target_k = 1_000
cwd = os.path.dirname(__file__)
queries_file: str = os.path.join(cwd, "queries.json")
vector_field: str = "vector"
self._knn_vector_store: KnnVectorStore = KnnVectorStore.get_instance(queries_file, vector_field)
self._queries_file: str = os.path.join(cwd, "queries.json")
self._vector_field: str = "vector"

def partition(self, partition_index, total_partitions):
return self
Expand All @@ -169,7 +168,7 @@ def params(self):
"size": self._params.get("k", 10),
"num_candidates": self._params.get("num-candidates", 100),
"target_k": self._target_k,
"knn_vector_store": self._knn_vector_store,
"knn_vector_store": KnnVectorStore.get_instance(self._queries_file, self._vector_field),
}


Expand Down

0 comments on commit faf17fd

Please sign in to comment.