fix: reshape matrix (#519)

fixes: #518
explodinggradients · Jan 26, 2024 · 594871f · 594871f
1 parent a76e5f0
commit 594871f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ragas/metrics/_answer_relevance.py b/src/ragas/metrics/_answer_relevance.py
@@ -85,7 +85,7 @@ def calculate_similarity(
         question_vec = np.asarray(self.embeddings.embed_query(question)).reshape(1, -1)
         gen_question_vec = np.asarray(
             self.embeddings.embed_documents(generated_questions)
-        )
+        ).reshape(len(generated_questions), -1)
         norm = np.linalg.norm(gen_question_vec, axis=1) * np.linalg.norm(
             question_vec, axis=1
         )