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

Add expr_str to return values in countdown.py

Browse source 
Modified return statement to include the modified expression string.
This commit is contained in:
SII-Whereby 2025-11-30 22:12:33 +08:00 • committed by GitHub
parent 85629bdfec
commit 7b6283e81b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
reasoning_gym/games/countdown.py
View file

@ -178,7 +178,7 @@ class CountdownDataset(ProceduralDataset):
for sym_name, num_str in replacements: for sym_name, num_str in replacements:
expr_str = expr_str.replace(sym_name, num_str) expr_str = expr_str.replace(sym_name, num_str)
return expr, numbers, syms return expr, numbers, syms, expr_str
def _generate_expression(self, rng: Random) -> tuple[str, list[int], int]: def _generate_expression(self, rng: Random) -> tuple[str, list[int], int]:
"""Generate a valid expression and its result """Generate a valid expression and its result