From 74d8d69bd42c253c255dc69904ee1fbd1eece0cf Mon Sep 17 00:00:00 2001 From: Sam Shleifer Date: Mon, 28 Sep 2020 23:20:03 -0400 Subject: [PATCH] [s2s] consistent output format across eval scripts (#7435) --- examples/seq2seq/rouge_cli.py | 2 +- examples/seq2seq/run_eval.py | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/examples/seq2seq/rouge_cli.py b/examples/seq2seq/rouge_cli.py index b193581bc8..6a54a72eb5 100644 --- a/examples/seq2seq/rouge_cli.py +++ b/examples/seq2seq/rouge_cli.py @@ -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 diff --git a/examples/seq2seq/run_eval.py b/examples/seq2seq/run_eval.py index 09830b66db..910d430bdd 100755 --- a/examples/seq2seq/run_eval.py +++ b/examples/seq2seq/run_eval.py @@ -152,8 +152,7 @@ def run_generate(verbose=True): print(scores) if args.score_path is not None: - path = args.score_path - json.dump(scores, open(path, "w")) + json.dump(scores, open(args.score_path, "w")) return scores