From 4bb299424099190e6e833746bb74a34c9eb9361a Mon Sep 17 00:00:00 2001
From: Sheharyar Ahmad <sheharyar.572@gmail.com>
Date: Fri, 29 Nov 2024 18:59:44 +0500
Subject: [PATCH] fix: invalid value for --max-num-levels when using CLI.

---
 vectordb_bench/backend/clients/alloydb/cli.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/vectordb_bench/backend/clients/alloydb/cli.py b/vectordb_bench/backend/clients/alloydb/cli.py
index aa3a5cbe..54d9b9fa 100644
--- 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
         )