Allow extra args on pretrain and debug_data

This commit is contained in:
Ines Montani 2020-07-11 19:17:59 +02:00
parent 70abcca60e
commit 5e683d03fe
2 changed files with 8 additions and 2 deletions

View File

@ -24,7 +24,10 @@ BLANK_MODEL_MIN_THRESHOLD = 100
BLANK_MODEL_THRESHOLD = 2000 BLANK_MODEL_THRESHOLD = 2000
@app.command("debug-data") @app.command(
"debug-data",
context_settings={"allow_extra_args": True, "ignore_unknown_options": True},
)
def debug_data_cli( def debug_data_cli(
# fmt: off # fmt: off
ctx: typer.Context, # This is only used to read additional arguments ctx: typer.Context, # This is only used to read additional arguments

View File

@ -24,7 +24,10 @@ from ..attrs import ID, HEAD
from .. import util from .. import util
@app.command("pretrain") @app.command(
"pretrain",
context_settings={"allow_extra_args": True, "ignore_unknown_options": True},
)
def pretrain_cli( def pretrain_cli(
# fmt: off # fmt: off
ctx: typer.Context, # This is only used to read additional arguments ctx: typer.Context, # This is only used to read additional arguments