fix: invalid value for --max-num-levels when using CLI.

zilliztech · Dec 2, 2024 · 4bb2994 · 4bb2994
1 parent 340eb41
commit 4bb2994
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vectordb_bench/backend/clients/alloydb/cli.py b/vectordb_bench/backend/clients/alloydb/cli.py
@@ -106,7 +106,7 @@ class AlloyDBScaNNTypedDict(AlloyDBTypedDict):
         int,
         click.option(
             "--max-num-levels",
-            type=click.Choice([1, 2]),
+            type=click.Choice(["1", "2"]),
             help="Maximum number of levels",
             default=1
         )