Skip to content

GitLab

Commit c490f5ce authored by thomwolf's avatar thomwolf
Browse files

added generation examples in tests

parent 8bb02c27
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
examples/test_examples.py
View file @ c490f5ce
...@@ -98,12 +98,12 @@ class ExamplesTests(unittest.TestCase): ...@@ -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()
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment