diff --git a/reasoning_gym/algorithmic/ransom_note.py b/reasoning_gym/algorithmic/ransom_note.py index 90f30758..633cb62c 100644 --- a/reasoning_gym/algorithmic/ransom_note.py +++ b/reasoning_gym/algorithmic/ransom_note.py @@ -112,7 +112,7 @@ class RansomNoteDataset(ProceduralDataset): return 0.0 s_answer = answer.strip() - if not s_answer == str(entry['answer']): + if not s_answer == str(entry["answer"]): return 0.01 else: return 1.0 diff --git a/tests/test_ransom_note.py b/tests/test_ransom_note.py index d2509955..f452ca2e 100644 --- a/tests/test_ransom_note.py +++ b/tests/test_ransom_note.py @@ -85,7 +85,7 @@ def test_group_anagrams_dataset_items(): assert solution == solvable # Test the scoring - assert dataset.score_answer(answer=item['answer'], entry=item) == 1.0 + assert dataset.score_answer(answer=item["answer"], entry=item) == 1.0 assert dataset.score_answer(answer="gibberish", entry=item) == 0.01 assert dataset.score_answer(answer=None, entry=item) == 0.0