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

fix run_generation test

Browse Source
This commit is contained in:
thomwolf
2019-07-15 17:52:35 +02:00
parent e691fc0963
commit 76da9765b6
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/test_examples.py
View File

@@ -103,7 +103,7 @@ class ExamplesTests(unittest.TestCase):
"--seed=42"] "--seed=42"]
model_type, model_name = ("--model_type=openai-gpt", model_type, model_name = ("--model_type=openai-gpt",
"--model_name_or_path=openai-gpt") "--model_name_or_path=openai-gpt")
with patch.object(sys, 'argv', testargs + [model_name]): with patch.object(sys, 'argv', testargs + [model_type, model_name]):
result = run_generation.main() result = run_generation.main()
self.assertGreaterEqual(len(result), 10) self.assertGreaterEqual(len(result), 10)