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

Fix #6428 (#6437)

Browse Source
This commit is contained in:
Sylvain Gugger
2020-08-12 08:47:30 -04:00
committed by GitHub
parent a8db954cda
commit 96c3329f19
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/transformers/hf_argparser.py
View File

@@ -4,7 +4,7 @@ import sys
from argparse import ArgumentParser from argparse import ArgumentParser
from enum import Enum from enum import Enum
from pathlib import Path from pathlib import Path
from typing import Any, Iterable, List, NewType, Tuple, Union from typing import Any, Iterable, List, NewType, Optional, Tuple, Union
DataClass = NewType("DataClass", Any) DataClass = NewType("DataClass", Any)
@@ -64,7 +64,7 @@ class HfArgumentParser(ArgumentParser):
kwargs["type"] = field.type kwargs["type"] = field.type
if field.default is not dataclasses.MISSING: if field.default is not dataclasses.MISSING:
kwargs["default"] = field.default kwargs["default"] = field.default
elif field.type is bool: elif field.type is bool or field.type is Optional[bool]:
kwargs["action"] = "store_false" if field.default is True else "store_true" kwargs["action"] = "store_false" if field.default is True else "store_true"
if field.default is True: if field.default is True:
field_name = f"--no-{field.name}" field_name = f"--no-{field.name}"