transpose

signwriting_similarity.py

@@ -134,13 +134,8 @@ class SignWritingSimilarity(evaluate.Metric):
         )
 
     def _compute(self, predictions, references):
-        # if only one reference is provided make sure we still use list of lists
-        if isinstance(references[0], str):
-            references = [references]
-        # else:
-        #     # transpose references, to be array of arrays
-        #     # the internal array is as long as the predictions, the external one is for multiple references.
-        #     references = list(zip(*references))
+        # the internal array is as long as the predictions, the external one is for multiple references.
+        references = list(zip(*references))
 
         score = self.metric.corpus_score(predictions, references)