[s2s] run_eval.py parses generate_kwargs (#6948)

2020-09-04 14:19:31 -04:00
parent 6078b12098
commit a4fc0c80b1
3 changed files with 36 additions and 24 deletions
--- a/examples/seq2seq/test_seq2seq_examples.py
+++ b/examples/seq2seq/test_seq2seq_examples.py
@@ -300,6 +300,10 @@ def test_run_eval(model):
        score_path,
        "--task",
        task,
+        "--num_beams",
+        "2",
+        "--length_penalty",
+        "2.0",
    ]
    with patch.object(sys, "argv", testargs):
        run_generate()