We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3e2e1d5 commit a248d26Copy full SHA for a248d26
src/svsbench/generate_ground_truth.py
@@ -27,7 +27,6 @@ def _read_args(argv: list[str] | None = None) -> argparse.Namespace:
27
"--distance",
28
help="Distance",
29
choices=tuple(consts.STR_TO_DISTANCE.keys()),
30
- type=consts.STR_TO_DISTANCE.get,
31
default="mip",
32
)
33
parser.add_argument(
@@ -50,7 +49,7 @@ def main(argv: str | None = None) -> None:
50
49
generate_ground_truth(
51
vecs_path=args.vecs_file,
52
query_file=args.query_file,
53
- distance=args.distance,
+ distance=consts.STR_TO_DISTANCE[args.distance],
54
num_vectors=args.num_vectors,
55
k=args.k,
56
num_threads=args.max_threads,
0 commit comments