From faf17fd7b1a9caf653c53919dc4fcd28cdc90d65 Mon Sep 17 00:00:00 2001 From: Panagiotis Bailis Date: Fri, 24 Nov 2023 18:32:50 +0200 Subject: [PATCH] minor refactoring --- dense_vector/track.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/dense_vector/track.py b/dense_vector/track.py index a625547b..386fd5ba 100644 --- a/dense_vector/track.py +++ b/dense_vector/track.py @@ -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 @@ -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), }