Skip to content

Commit 57468c1

Browse files
committed
Allow CLI arguments to be set through ENV
1 parent 15fd7f5 commit 57468c1

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/guidellm/__main__.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,8 @@ def cli():
2626

2727

2828
@cli.command(
29-
help="Run a benchmark against a generative model using the specified arguments."
29+
help="Run a benchmark against a generative model using the specified arguments.",
30+
context_settings={"auto_envvar_prefix": "GUIDELLM"},
3031
)
3132
@click.option(
3233
"--scenario",
@@ -339,7 +340,8 @@ def preprocess():
339340
"Convert a dataset to have specific prompt and output token sizes.\n"
340341
"DATA: Path to the input dataset or dataset ID.\n"
341342
"OUTPUT_PATH: Path to save the converted dataset, including file suffix."
342-
)
343+
),
344+
context_settings={"auto_envvar_prefix": "GUIDELLM"},
343345
)
344346
@click.argument(
345347
"data",

0 commit comments

Comments
 (0)