update QA models tests + run_generation

This commit is contained in:
thomwolf committed 2019-07-15 17:45:24 +02:00
1 parent 15d8b1266c
commit e691fc0963
4 files changed
+41 -27

No files matched your search

+2 -1
View File
@@ -101,7 +101,8 @@ class ExamplesTests(unittest.TestCase):
"--prompt=Hello",
"--length=10",
"--seed=42"]
model_name = "--model_name=openai-gpt"
model_type, model_name = ("--model_type=openai-gpt",
"--model_name_or_path=openai-gpt")
with patch.object(sys, 'argv', testargs + [model_name]):
result = run_generation.main()
self.assertGreaterEqual(len(result), 10)