From 76da9765b6c9acc521bf589fac81075aabe50d78 Mon Sep 17 00:00:00 2001 From: thomwolf Date: Mon, 15 Jul 2019 17:52:35 +0200 Subject: [PATCH] fix run_generation test --- examples/test_examples.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/test_examples.py b/examples/test_examples.py index 2f88d129f8..688401ebc9 100644 --- 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)