Skip to content

Commit a248d26

Browse files
authored
Fix generate_ground_truth distance argument (#10)
1 parent 3e2e1d5 commit a248d26

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

src/svsbench/generate_ground_truth.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,6 @@ def _read_args(argv: list[str] | None = None) -> argparse.Namespace:
2727
"--distance",
2828
help="Distance",
2929
choices=tuple(consts.STR_TO_DISTANCE.keys()),
30-
type=consts.STR_TO_DISTANCE.get,
3130
default="mip",
3231
)
3332
parser.add_argument(
@@ -50,7 +49,7 @@ def main(argv: str | None = None) -> None:
5049
generate_ground_truth(
5150
vecs_path=args.vecs_file,
5251
query_file=args.query_file,
53-
distance=args.distance,
52+
distance=consts.STR_TO_DISTANCE[args.distance],
5453
num_vectors=args.num_vectors,
5554
k=args.k,
5655
num_threads=args.max_threads,

0 commit comments

Comments
 (0)