mirror of
https://github.com/open-thought/reasoning-gym.git
synced 2026-04-19 12:58:07 +00:00
adapt unit tests to partial match changes
This commit is contained in:
parent
0169c0fba4
commit
044da9687f
2 changed files with 4 additions and 2 deletions
|
|
@ -98,7 +98,7 @@ def test_arc_1d_scoring():
|
|||
assert dataset.score_answer(entry["answer"], entry) == 1.0
|
||||
|
||||
# Test partial match (answer contained within response)
|
||||
assert dataset.score_answer(f"The answer is: {entry['answer']}", entry) == 0.5
|
||||
assert dataset.score_answer(f"The answer is: {entry['answer']}", entry) > 0.5
|
||||
|
||||
# Test incorrect answer
|
||||
assert dataset.score_answer("wrong answer", entry) == 0.01
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue