Skip to content

Commit e1e04bb

Browse files
kahnemgaido91
authored andcommitted
fix SacrebleuScorer.score()
Summary: fix SacrebleuScorer.score() Reviewed By: sravyapopuri388 Differential Revision: D33311843 fbshipit-source-id: 8536baceab6ef2e7c9c4a9a8a005abaa6a9229f0
1 parent 901a143 commit e1e04bb

File tree

1 file changed

+8
-7
lines changed

1 file changed

+8
-7
lines changed

fairseq/scoring/bleu.py

+8-7
Original file line numberDiff line numberDiff line change
@@ -59,16 +59,17 @@ def add_string(self, ref, pred):
5959
self.ref.append(self.tokenizer.tokenize(ref))
6060
self.pred.append(self.tokenizer.tokenize(pred))
6161

62-
def score(self, order=4):
63-
return self.result_string(order).score
64-
65-
def result_string(self, order=4):
62+
def _score(self, order=4):
6663
if order != 4:
6764
raise NotImplementedError
6865
# tokenization and lowercasing are performed by self.tokenizer instead.
69-
return self.sacrebleu.corpus_bleu(
70-
self.pred, [self.ref], tokenize="none"
71-
).format()
66+
return self.sacrebleu.corpus_bleu(self.pred, [self.ref], tokenize="none")
67+
68+
def score(self, order=4):
69+
return self._score(order).score
70+
71+
def result_string(self, order=4):
72+
return self._score(order).format()
7273

7374

7475
@dataclass

0 commit comments

Comments
 (0)