Fix test

langchain-ai · Nov 12, 2024 · 884476d · 884476d
1 parent 70c6dc9
commit 884476d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/python/tests/unit_tests/evaluation/test_runner.py b/python/tests/unit_tests/evaluation/test_runner.py
@@ -80,7 +80,9 @@ def request(self, verb: str, endpoint: str, *args, **kwargs):
                 res = MagicMock()
                 res.json.return_value = {
                     "runs": [
-                        r for r in self.runs.values() if "reference_example_id" in r
+                        r
+                        for r in self.runs.values()
+                        if r["trace_id"] == r["id"] and r.get("reference_example_id")
                     ]
                 }
                 return res