riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 33 Packages Projects Releases Wiki Activity

added generation examples in tests

Browse Source
This commit is contained in:
thomwolf 2019-07-13 15:26:58 +02:00
parent 8bb02c27e2
commit c490f5ce87
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
examples/test_examples.py
View File

@ -98,12 +98,12 @@ class ExamplesTests(unittest.TestCase):
testargs = ["run_generation.py",
"--prompt=Hello",
"--length=10",
"--seed=42"]
model_name = "--model_name=openai-gpt"
with patch.object(sys, 'argv', testargs + [model_name]):
result = run_generation.main()
self.assertGreaterEqual(result['f1'], 30)
self.assertGreaterEqual(result['exact'], 30)
self.assertGreaterEqual(len(result), 10)
if __name__ == "__main__":
unittest.main()