Skip to content

Commit 75bbdfb

Browse files
authored
Add --task-label option for cli (zilliztech#517)
* Add --task-label option for cli * Fix lint issues
1 parent 0122126 commit 75bbdfb

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

vectordb_bench/cli/cli.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -405,6 +405,7 @@ class CommonTypedDict(TypedDict):
405405
show_default=True,
406406
),
407407
]
408+
task_label: Annotated[str, click.option("--task-label", help="Task label")]
408409

409410

410411
class HNSWBaseTypedDict(TypedDict):
@@ -499,10 +500,11 @@ def run(
499500
parameters["search_concurrent"],
500501
),
501502
)
503+
task_label = parameters["task_label"]
502504

503505
log.info(f"Task:\n{pformat(task)}\n")
504506
if not parameters["dry_run"]:
505-
benchmark_runner.run([task])
507+
benchmark_runner.run([task], task_label)
506508
time.sleep(5)
507509
if global_result_future:
508510
wait([global_result_future])

0 commit comments

Comments
 (0)