parse arguments from dict (#4869)

* add parse_dict to parse arguments from dict

* add unit test for parse_dict
This commit is contained in:
Suraj Patil
2020-07-31 14:14:23 +05:30
committed by GitHub
parent cf3cf304ca
commit 838dc06ff5
2 changed files with 27 additions and 0 deletions

View File

@@ -152,6 +152,20 @@ class HfArgumentParserTest(unittest.TestCase):
args = parser.parse_args("--foo 12 --bar 3.14 --baz 42 --ces a b c --des 1 2 3".split())
self.assertEqual(args, Namespace(foo=12, bar=3.14, baz="42", ces=["a", "b", "c"], des=[1, 2, 3]))
def test_parse_dict(self):
parser = HfArgumentParser(BasicExample)
args_dict = {
"foo": 12,
"bar": 3.14,
"baz": "42",
"flag": True,
}
parsed_args = parser.parse_dict(args_dict)[0]
args = BasicExample(**args_dict)
self.assertEqual(parsed_args, args)
def test_integration_training_args(self):
parser = HfArgumentParser(TrainingArguments)
self.assertIsNotNone(parser)