open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-28 17:29:39 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

lint

Browse source
This commit is contained in:
Rich Jones 2025-02-14 12:17:50 +01:00
parent aae6a58e35
commit 9c12701e7a
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
reasoning_gym/algorithmic/ransom_note.py
View file

@ -112,7 +112,7 @@ class RansomNoteDataset(ProceduralDataset):
return 0.0 return 0.0
s_answer = answer.strip() s_answer = answer.strip()
if not s_answer == str(entry['answer']): if not s_answer == str(entry["answer"]):
return 0.01 return 0.01
else: else:
return 1.0 return 1.0

2
tests/test_ransom_note.py
View file

@ -85,7 +85,7 @@ def test_group_anagrams_dataset_items():
assert solution == solvable assert solution == solvable
# Test the scoring # 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="gibberish", entry=item) == 0.01
assert dataset.score_answer(answer=None, entry=item) == 0.0 assert dataset.score_answer(answer=None, entry=item) == 0.0