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

Smarter prediction loop and no- -> no_ in console args (#8151)

Browse Source 
* Smarter prediction loop and no- -> no_ in console args

* Fix test
This commit is contained in:
Sylvain Gugger
2020-10-29 10:56:25 -04:00
committed by GitHub
parent b0f1c0ee30
commit acf56408d8
5 changed files with 22 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_hf_argparser.py
View File

@@ -93,13 +93,13 @@ class HfArgumentParserTest(unittest.TestCase):
expected = argparse.ArgumentParser()
expected.add_argument("--foo", action="store_true")
expected.add_argument("--no-baz", action="store_false", dest="baz")
expected.add_argument("--no_baz", action="store_false", dest="baz")
self.argparsersEqual(parser, expected)
args = parser.parse_args([])
self.assertEqual(args, Namespace(foo=False, baz=True))
args = parser.parse_args(["--foo", "--no-baz"])
args = parser.parse_args(["--foo", "--no_baz"])
self.assertEqual(args, Namespace(foo=True, baz=False))
def test_with_enum(self):