AmitMY commited on
Commit
c6e9951
·
1 Parent(s): de3203d
Files changed (1) hide show
  1. signwriting_similarity.py +4 -0
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}