Add an option to HfArgumentParser.parse_{dict,json_file} to raise an Exception when there extra keys (#18692)

* Update parser to track unneeded keys, off by default

* Fix formatting

* Fix docstrings and defaults in HfArgparser

* Fix formatting
This commit is contained in:
Felix Schneider
2022-08-31 20:26:45 +02:00
committed by GitHub
parent f210e2a414
commit 86387fe87f
2 changed files with 48 additions and 4 deletions

View File

@@ -245,6 +245,19 @@ class HfArgumentParserTest(unittest.TestCase):
args = BasicExample(**args_dict)
self.assertEqual(parsed_args, args)
def test_parse_dict_extra_key(self):
parser = HfArgumentParser(BasicExample)
args_dict = {
"foo": 12,
"bar": 3.14,
"baz": "42",
"flag": True,
"extra": 42,
}
self.assertRaises(ValueError, parser.parse_dict, args_dict, allow_extra_keys=False)
def test_integration_training_args(self):
parser = HfArgumentParser(TrainingArguments)
self.assertIsNotNone(parser)