Fix test ExamplesTests::test_run_translation (huggingface#30281)

fix Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
zucchini-nlp · Apr 18, 2024 · 0e4320d · 0e4320d
1 parent 467825d
commit 0e4320d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/examples/pytorch/test_pytorch_examples.py b/examples/pytorch/test_pytorch_examples.py
@@ -372,6 +372,7 @@ def test_run_translation(self):
             --predict_with_generate
             --source_lang en_XX
             --target_lang ro_RO
+            --max_source_length 512
         """.split()
 
         with patch.object(sys, "argv", testargs):