fix naming

a38a0450 · lintangsutawika · 1cc2a764 · a38a0450
Commit a38a0450 authored Aug 21, 2023 by lintangsutawika
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 4 deletions

lm_eval/tasks/wmt2016/metrics.py lm_eval/tasks/wmt2016/metrics.py +3 -4

No files found.
--- a/lm_eval/tasks/wmt2016/metrics.py
+++ b/lm_eval/tasks/wmt2016/metrics.py
 import evaluate

-def bleu(predictions, references):
-    rouge_fn = evaluate.load('bleu')
-    results = rouge_fn.compute(predictions=predictions, references=references)
-    return results['bleu']
+def bleu(references, predictions):
+    bleu_fn = evaluate.load('bleu')
+    return bleu_fn.compute(predictions=predictions, references=references)