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 #484] resolve basic_arithmetic fails when size is large (#485)

Browse source 
* [fix] resolve basic_arithmetic fails when size is large by replacing zero divisor with 1
This commit is contained in:
theblackcat102 2025-07-07 16:46:23 +08:00 • committed by GitHub
parent bf451d5197
commit 2d19f13e0f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 19 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
reasoning_gym/arithmetic/basic_arithmetic.py
View file

@ -191,6 +191,9 @@ class BasicArithmeticDataset(ProceduralDataset):
space_parts.append(" ")
space_parts.append(p)
expression = "".join(space_parts).strip()
# Avoid division-by-zero in final evaluation by converting '/0' patterns to '/1'
if "/ 0" in expression:
expression = expression.replace("/ 0", "/ 1")
result = eval_floordiv(expression) # Note: eval is safe here as we control the input
return expression, result