Add default value for pre_reordering_num_neighbors in cli options.

zilliztech · Nov 28, 2024 · e67a0a6 · e67a0a6
1 parent 97f2688
commit e67a0a6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/vectordb_bench/backend/clients/alloydb/cli.py b/vectordb_bench/backend/clients/alloydb/cli.py
@@ -70,8 +70,8 @@ class AlloyDBScaNNTypedDict(AlloyDBTypedDict):
         click.option("--num-leaves-to-search", type=int, help="Number of leaves to search", required=True)
     ]
     pre_reordering_num_neighbors: Annotated[
-        Optional[int],
-        click.option("--pre-reordering-num-neighbors", type=int, help="Pre-reordering number of neighbors",)
+        int,
+        click.option("--pre-reordering-num-neighbors", type=int, help="Pre-reordering number of neighbors", default=200)
     ]
     max_top_neighbors_buffer_size: Annotated[
         int,