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

extract answer from last answer tag

Browse source
This commit is contained in:
Andreas Koepf 2025-01-28 16:37:19 +00:00
parent cc0312e446
commit c196d622e0
5 changed files with 31 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

14
reasoning_gym/utils.py
View file

@ -12,11 +12,13 @@ The assistant first thinks about the reasoning process in the mind and then prov
def extract_answer(completion: str, tag_name: str = "answer") -> Optional[str]:
regex = f"<{tag_name}>(.*?)</{tag_name}>"
answer_match = re.search(
regex,
completion,
flags=re.DOTALL,
matches = list(
re.finditer(
regex,
completion,
flags=re.DOTALL,
)
)
if not answer_match:
if not matches:
return None
return answer_match.group(1)
return matches[-1].group(1)