Fix CI failing

deepset-ai · Apr 3, 2024 · 4542897 · 4542897
1 parent e642081
commit 4542897
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/haystack/components/evaluators/sas_evaluator.py b/haystack/components/evaluators/sas_evaluator.py
@@ -178,7 +178,9 @@ def run(self, ground_truths_answers: List[str], predicted_answers: List[str]) ->
             )
 
             # Compute cosine-similarities
-            similarity_scores = [util.cos_sim(p, l).cpu() for p, l in zip(predictions_embeddings, label_embeddings)]
+            similarity_scores = [
+                util.cos_sim(p, l).cpu().numpy() for p, l in zip(predictions_embeddings, label_embeddings)
+            ]
 
         sas_score = np_mean(similarity_scores)