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 script to check collision and fix bug in basic_arithmetic dataset

Browse source
This commit is contained in:
abdulhakeem 2025-03-31 08:39:40 -05:00
parent 695aad4dbc
commit 25a6c22695
8 changed files with 463 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

11
reasoning_gym/arithmetic/basic_arithmetic.py
View file

@ -161,11 +161,14 @@ class BasicArithmeticDataset(ProceduralDataset):
right_parts.append(")")
else:
divisor = rng.choice(find_common_divisors(dividend, 0))
left_parts.append(str(divisor))
left_parts.append("+")
if dividend != 0:
divisor = rng.choice(find_common_divisors(dividend, 0))
else:
divisor = rng.randint(1, 10**num_digits - 1)
left_parts.append(str(divisor))
left_parts.append("+")
left_parts.extend(right_parts)
left_parts.extend(right_parts)
else:
if dividend != 0:
divisor = rng.choice(find_common_divisors(dividend, 0))