transZ commited on
Commit
e29d4b0
1 Parent(s): 18229bd
Files changed (1) hide show
  1. sbert_cosine.py +1 -1
sbert_cosine.py CHANGED
@@ -131,7 +131,7 @@ class sbert_cosine(evaluate.Metric):
131
  # avg = lambda x: sum(x) / len(x)
132
 
133
  with torch.no_grad():
134
- scores = [calculate(pred, ref) for pred, ref in zip(predictions, references]
135
 
136
  return {
137
  "score": scores,
 
131
  # avg = lambda x: sum(x) / len(x)
132
 
133
  with torch.no_grad():
134
+ scores = [calculate(pred, ref) for pred, ref in zip(predictions, references)]
135
 
136
  return {
137
  "score": scores,