Fix test ExamplesTests::test_run_translation (#30281)

fix Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2024-04-17 11:46:33 +02:00
parent 304c6a1e0d
commit 05dab4e5ba
1 changed files with 1 additions and 0 deletions
--- a/examples/pytorch/test_pytorch_examples.py
+++ b/examples/pytorch/test_pytorch_examples.py
@@ -372,6 +372,7 @@ class ExamplesTests(TestCasePlus):
            --predict_with_generate
            --source_lang en_XX
            --target_lang ro_RO
+            --max_source_length 512
        """.split()

        with patch.object(sys, "argv", testargs):