Fixed bug in comet score.

community_tasks/swiss_legal_evals.py

@@ -593,7 +593,10 @@ def compute(
         predictions = [response[0].result for response in responses]
         sources = [kwargs["formatted_doc"].specific["source"] for kwargs["formatted_doc"] in formatted_docs]
 
-        data = [{"src": src, "mt": pred, "ref": gold} for src, pred, gold in zip(sources, predictions, golds)]
+        data = [
+            {"src": src, "mt": pred if isinstance(pred, str) else pred[0], "ref": gold}
+            for src, pred, gold in zip(sources, predictions, golds)
+        ]
         model_output = self.model.predict(
             data,
             batch_size=self.batch_size,