open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-19 12:58:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix bug in normalize_answer method (#444)

Browse source
This commit is contained in:
Adefioye 2025-06-02 01:58:54 -05:00 • committed by GitHub
parent c0e98f93b4
commit 9053009dbe
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 8 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
reasoning_gym/arithmetic/prime_factorization.py
View file

@ -49,7 +49,10 @@ class PrimeFactorizationDataset(ProceduralDataset):
def _normalize_answer(self, answer: str) -> list[int]:
"""Parse and sort factors from a string"""
return sorted([int(factor.strip()) for factor in answer.split("×")])
if not answer or answer.strip() == "":
return []
return sorted([int(factor.strip()) for factor in answer.split("×") if factor.strip() != ""])
def score_answer(self, answer: Optional[str], entry: dict[str, Any]) -> float:
oracle_answer = entry["answer"]