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

Fix PoolMatrixConfigs::score_answer(), add unit tests (#215)

Browse source
This commit is contained in:
Andreas Köpf 2025-02-26 00:43:18 +01:00 • committed by GitHub
parent 99fec3425d
commit 48f082663a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 27 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
reasoning_gym/algorithmic/pool_matrix.py
View file

@ -102,8 +102,8 @@ class PoolMatrixDataset(ProceduralDataset):
reward = 0.0
try:
oracle_answer = np.array(entry["answer"])
answer = np.array(answer)
oracle_answer = np.loadtxt(entry["answer"].splitlines(), dtype=np.float32)
answer = np.loadtxt(answer.splitlines(), dtype=np.float32)
if oracle_answer.shape == answer.shape and np.allclose(oracle_answer, answer):
reward = 1.0
elif oracle_answer.shape == answer.shape: