From 96c3329f19f28e47eab7f9f20ed3504619e16722 Mon Sep 17 00:00:00 2001 From: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> Date: Wed, 12 Aug 2020 08:47:30 -0400 Subject: [PATCH] Fix #6428 (#6437) --- src/transformers/hf_argparser.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/transformers/hf_argparser.py b/src/transformers/hf_argparser.py index 6c4e3f204b..af350c1128 100644 --- a/src/transformers/hf_argparser.py +++ b/src/transformers/hf_argparser.py @@ -4,7 +4,7 @@ import sys from argparse import ArgumentParser from enum import Enum 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) @@ -64,7 +64,7 @@ class HfArgumentParser(ArgumentParser): kwargs["type"] = field.type if field.default is not dataclasses.MISSING: 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" if field.default is True: field_name = f"--no-{field.name}"