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

Data collisions notebooks and data (#406)

Browse source 
* Add collisions data

* Fix logic issues in basic_arithmetic and gsm_symbolic data
This commit is contained in:
Adefioye 2025-04-02 02:36:09 -05:00 • committed by GitHub
parent 43c739cb3e
commit 5b653b346c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 509 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
reasoning_gym/arithmetic/basic_arithmetic.py
View file

@ -161,10 +161,12 @@ class BasicArithmeticDataset(ProceduralDataset):
right_parts.append(")")
else:
divisor = rng.choice(find_common_divisors(dividend, 0))
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)
else:
if dividend != 0: