SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

examples/seq2seq/run_eval.py fixes and docs (#5322)

Browse Source
This commit is contained in:
Sam Shleifer
2020-06-26 19:20:43 -04:00
committed by GitHub
parent 5543b30aa6
commit 393b8dc09a
5 changed files with 79 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/seq2seq/test_seq2seq_examples.py
View File

@@ -198,7 +198,7 @@ def test_run_eval_bart(model):
assert not output_file_name.exists()
articles = [" New York (CNN)When Liana Barrientos was 23 years old, she got married in Westchester County."]
_dump_articles(input_file_name, articles)
testargs = ["run_eval.py", str(input_file_name), str(output_file_name), model] # TODO: test score_path
testargs = ["run_eval.py", model, str(input_file_name), str(output_file_name)] # TODO: test score_path
with patch.object(sys, "argv", testargs):
run_generate()
assert Path(output_file_name).exists()