[s2s] consistent output format across eval scripts (#7435)

This commit is contained in:
Sam Shleifer
2020-09-28 23:20:03 -04:00
committed by GitHub
parent 671b278e25
commit 74d8d69bd4
2 changed files with 2 additions and 3 deletions

View File

@@ -9,7 +9,7 @@ def calculate_rouge_path(pred_path, tgt_path, save_path=None, **kwargs):
tgt_lns = [x.strip() for x in open(tgt_path).readlines()][: len(pred_lns)]
metrics = calculate_rouge(pred_lns, tgt_lns, **kwargs)
if save_path is not None:
save_json(metrics, save_path)
save_json(metrics, save_path, indent=None)
return metrics # these print nicely