Spaces:
Sleeping
Sleeping
Fix bug
Browse files- 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,
|