Skip to content

Commit b13dd7b

Browse files
committed
Fix HNSW-PQ m parameter conflicts wioth M.
1 parent cda6227 commit b13dd7b

1 file changed

Lines changed: 9 additions & 0 deletions

File tree

  • vectordb_bench/backend/clients/milvus

vectordb_bench/backend/clients/milvus/cli.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -179,6 +179,14 @@ class MilvusRefineTypedDict(TypedDict):
179179

180180

181181
class MilvusHNSWPQTypedDict(CommonTypedDict, MilvusTypedDict, MilvusHNSWTypedDict, MilvusRefineTypedDict):
182+
pq_m: Annotated[
183+
int,
184+
click.option(
185+
"--pq-m",
186+
type=int,
187+
required=True,
188+
),
189+
]
182190
nbits: Annotated[
183191
int,
184192
click.option(
@@ -209,6 +217,7 @@ def MilvusHNSWPQ(**parameters: Unpack[MilvusHNSWPQTypedDict]):
209217
M=parameters["m"],
210218
efConstruction=parameters["ef_construction"],
211219
ef=parameters["ef_search"],
220+
m=parameters["pq_m"],
212221
nbits=parameters["nbits"],
213222
refine=parameters["refine"],
214223
refine_type=parameters["refine_type"],

0 commit comments

Comments
 (0)