fix run_generation test

2019-07-15 17:52:35 +02:00
parent e691fc0963
commit 76da9765b6
1 changed files with 1 additions and 1 deletions
--- a/examples/test_examples.py
+++ b/examples/test_examples.py
@@ -103,7 +103,7 @@ class ExamplesTests(unittest.TestCase):
                    "--seed=42"]
        model_type, model_name = ("--model_type=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()
            self.assertGreaterEqual(len(result), 10)