From c3dde0e392fcbd03cd8d362107e40c57a894d95c Mon Sep 17 00:00:00 2001 From: Dom <97384583+tosemml@users.noreply.github.com> Date: Tue, 8 Aug 2023 19:14:40 -0700 Subject: [PATCH] use np.add --- CodeT5/evaluator/smooth_bleu.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/CodeT5/evaluator/smooth_bleu.py b/CodeT5/evaluator/smooth_bleu.py index 76f19b1..a775569 100644 --- a/CodeT5/evaluator/smooth_bleu.py +++ b/CodeT5/evaluator/smooth_bleu.py @@ -20,6 +20,7 @@ import sys, math, re, xml.sax.saxutils import subprocess import os +import numpy as np # Added to bypass NIST-style pre-processing of hyp and ref files -- wade nonorm = 0 @@ -194,7 +195,7 @@ def bleuFromMaps(m1, m2): for key in m1: if key in m2: bl = bleu(m1[key], m2[key][0]) - score = [score[i] + bl[i] for i in range(0, len(bl))] + score = np.add(bl,score) num += 1 return [s * 100.0 / num for s in score]