SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Use with_extension to change the extension (#4203)

Browse Source 
As per https://github.com/huggingface/transformers/pull/3934#discussion_r421307659
This commit is contained in:
Bram Vanroy
2020-05-07 17:14:56 +02:00
committed by GitHub
parent ebf80e2e70
commit e4fd5e3999
1 changed files with 1 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/transformers/hf_argparser.py
View File

@@ -11,10 +11,6 @@ DataClass = NewType("DataClass", Any)
DataClassType = NewType("DataClassType", Any) DataClassType = NewType("DataClassType", Any)
def trim_suffix(s: str, suffix: str):
return s if not s.endswith(suffix) or len(suffix) == 0 else s[: -len(suffix)]
class HfArgumentParser(ArgumentParser): class HfArgumentParser(ArgumentParser):
""" """
This subclass of `argparse.ArgumentParser` uses type hints on dataclasses This subclass of `argparse.ArgumentParser` uses type hints on dataclasses
@@ -109,8 +105,7 @@ class HfArgumentParser(ArgumentParser):
(same as argparse.ArgumentParser.parse_known_args) (same as argparse.ArgumentParser.parse_known_args)
""" """
if look_for_args_file and len(sys.argv): if look_for_args_file and len(sys.argv):
basename = trim_suffix(sys.argv[0], ".py") args_file = Path(sys.argv[0]).with_suffix(".args")
args_file = Path(f"{basename}.args")
if args_file.exists(): if args_file.exists():
fargs = args_file.read_text().split() fargs = args_file.read_text().split()
args = fargs + args if args is not None else fargs + sys.argv[1:] args = fargs + args if args is not None else fargs + sys.argv[1:]