SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 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
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", testargs = ["run_generation.py",
"--prompt=Hello", "--prompt=Hello",
"--length=10",
"--seed=42"] "--seed=42"]
model_name = "--model_name=openai-gpt" model_name = "--model_name=openai-gpt"
with patch.object(sys, 'argv', testargs + [model_name]): with patch.object(sys, 'argv', testargs + [model_name]):
result = run_generation.main() result = run_generation.main()
self.assertGreaterEqual(result['f1'], 30) self.assertGreaterEqual(len(result), 10)
self.assertGreaterEqual(result['exact'], 30)
if __name__ == "__main__": if __name__ == "__main__":
unittest.main() unittest.main()