Spaces:
Sleeping
Sleeping
add list
Browse files
signwriting_similarity.py
CHANGED
@@ -134,6 +134,10 @@ class SignWritingSimilarity(evaluate.Metric):
|
|
134 |
)
|
135 |
|
136 |
def _compute(self, predictions, references):
|
|
|
|
|
|
|
|
|
137 |
score = self.metric.corpus_score(predictions, references)
|
138 |
|
139 |
return {"score": score}
|
|
|
134 |
)
|
135 |
|
136 |
def _compute(self, predictions, references):
|
137 |
+
# if only one reference is provided make sure we still use list of lists
|
138 |
+
if isinstance(references[0], str):
|
139 |
+
references = [[ref] for ref in references]
|
140 |
+
|
141 |
score = self.metric.corpus_score(predictions, references)
|
142 |
|
143 |
return {"score": score}
|