Skip to content

Commit 5297e8e

Browse files
authored
Merge pull request #1 from 1kbgz/tkp/hf
Allow argv passthrough
2 parents e04b5a7 + 1f103dd commit 5297e8e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

p2a/model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -322,9 +322,9 @@ def create_model_parser(model: "BaseModel") -> ArgumentParser:
322322
return parser
323323

324324

325-
def parse_extra_args_model(model: "BaseModel"):
325+
def parse_extra_args_model(model: "BaseModel", argv: List[str] = None) -> Union["BaseModel", Dict]:
326326
# Parse the extra args and update the model
327-
args, kwargs = parse_extra_args(create_model_parser(model))
327+
args, kwargs = parse_extra_args(create_model_parser(model), argv)
328328

329329
for key, value in args.items():
330330
# Handle nested fields

0 commit comments

Comments
 (0)