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
@app.command("debug-data")
@app.command(
"debug-data",
context_settings={"allow_extra_args": True, "ignore_unknown_options": True},
)
def debug_data_cli(
# fmt: off
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
@app.command("pretrain")
@app.command(
"pretrain",
context_settings={"allow_extra_args": True, "ignore_unknown_options": True},
)
def pretrain_cli(
# fmt: off
ctx: typer.Context, # This is only used to read additional arguments