fix: column rename (#1324)

explodinggradients · Sep 18, 2024 · 502b336 · 502b336
1 parent c40891b
commit 502b336
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/ragas/metrics/_noise_sensitivity.py b/src/ragas/metrics/_noise_sensitivity.py
@@ -44,7 +44,7 @@ class NoiseSensitivity(MetricWithLLM, SingleTurnMetric):
             MetricType.SINGLE_TURN: {
                 "user_input",
                 "response",
-                "ground_truth",
+                "reference",
                 "retrieved_contexts",
             }
         }
@@ -257,7 +257,7 @@ async def _ascore(self: t.Self, row: t.Dict, callbacks: Callbacks) -> float:
         answers["retrieved2ground_truth"] = np.array(gt_verdictslist).T
         answers["retrieved2answer"] = np.array(ans_verdictslist).T
         answers["ground_truth2answer"] = await self._evaluate_statement_faithfulness(
-            ans_statements, row["ground_truth"], callbacks
+            ans_statements, row["reference"], callbacks
         )
         answers["ground_truth2answer"] = np.array([answers["ground_truth2answer"]])
         answers = {k: v.astype(bool) for k, v in answers.items()}