Skip to content

Commit e212c05

Browse files
authored
Fix calibration query and distance (#8)
1 parent e9357bd commit e212c05

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/svsbench/build.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,7 @@ def main(argv: str | None = None) -> None:
115115
index, name, ingest_time, delete_time = build_dynamic(
116116
vecs_path=args.vecs_file,
117117
svs_type=args.svs_type,
118-
distance=args.distance,
118+
distance=consts.STR_TO_DISTANCE[args.distance],
119119
idx_dir=args.idx_dir,
120120
num_vectors=args.num_vectors,
121121
graph_max_degree=args.graph_max_degree,

src/svsbench/search.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -204,7 +204,7 @@ def search(
204204
for batch_size_idx, batch_size in enumerate(batch_sizes):
205205
index.num_threads = min(max_threads, batch_size)
206206
if search_window_sizes is None:
207-
if calibration_query_path is None:
207+
if calibration_query_path is not None:
208208
calibration_query = svs.read_vecs(str(calibration_query_path))
209209
if calibration_ground_truth_path is None:
210210
raise ValueError(

0 commit comments

Comments
 (0)