fix-lenshtein test

Agenta-AI · May 28, 2024 · 111c2e5 · 111c2e5
1 parent 3fba85d
commit 111c2e5
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/agenta-backend/agenta_backend/tests/unit/test_evaluators.py b/agenta-backend/agenta_backend/tests/unit/test_evaluators.py
@@ -180,37 +180,37 @@ def test_auto_contains_json(output, expected):
         (
             "hello world",
             {"correct_answer": "hello world"},
-            {"threshold": 5, "correct_answer_keys": ["correct_answer"]},
+            {"threshold": 5, "correct_answer_key": "correct_answer"},
             True,
         ),
         (
             "hello world",
             {"correct_answer": "hola mundo"},
-            {"threshold": 5, "correct_answer_keys": ["correct_answer"]},
+            {"threshold": 5, "correct_answer_key": "correct_answer"},
             False,
         ),
         (
             "hello world",
             {"correct_answer": "hello world!"},
-            {"threshold": 2, "correct_answer_keys": ["correct_answer"]},
+            {"threshold": 2, "correct_answer_key": "correct_answer"},
             True,
         ),
         (
             "hello world",
             {"correct_answer": "hello wor"},
-            {"threshold": 10, "correct_answer_keys": ["correct_answer"]},
+            {"threshold": 10, "correct_answer_key": "correct_answer"},
             True,
         ),
         (
             "hello world",
             {"correct_answer": "hello worl"},
-            {"correct_answer_keys": ["correct_answer"]},
+            {"correct_answer_key": "correct_answer"},
             1,
         ),
         (
             "hello world",
             {"correct_answer": "helo world"},
-            {"correct_answer_keys": ["correct_answer"]},
+            {"correct_answer_key": "correct_answer"},
             1,
         ),
     ],