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 4ae3685 commit a52fcf6Copy full SHA for a52fcf6
src/svsbench/build.py
@@ -42,7 +42,6 @@ def _read_args(argv: list[str] | None = None) -> argparse.Namespace:
42
"--distance",
43
choices=tuple(consts.STR_TO_DISTANCE.keys()),
44
default="mip",
45
- type=consts.STR_TO_DISTANCE.get,
46
)
47
parser.add_argument(
48
"--max_threads_init",
@@ -103,7 +102,7 @@ def main(argv: str | None = None) -> None:
103
102
index, name = build_static(
104
vecs_path=args.vecs_file,
105
svs_type=args.svs_type,
106
- distance=args.distance,
+ distance=consts.STR_TO_DISTANCE[args.distance],
107
graph_max_degree=args.graph_max_degree,
108
window_size=args.window_size,
109
prune_to=args.prune_to,
0 commit comments