add fix for matching elements

meaningbert.py

@@ -14,6 +14,7 @@
 """ MeaningBERT metric. """
 
 from contextlib import contextmanager
+from itertools import chain
 from typing import List, Dict
 
 import datasets
@@ -118,6 +119,7 @@ class MeaningBERT(evaluate.Metric):
         ), "The number of document is different of the number of simplifications."
         hashcode = _HASH
 
+        # Index of sentence with perfect match between two sentences
         matching_index = [
             i for i, item in enumerate(documents) if item in simplifications
         ]
@@ -146,6 +148,10 @@ class MeaningBERT(evaluate.Metric):
 
         scores = scores.logits.tolist()
 
+        # Flatten the list of list of logits
+        scores = list(chain(*scores))
+
+        # Handle case of perfect match
         if len(matching_index) > 0:
             for matching_element_index in matching_index:
                 scores[matching_element_index] = 100