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

formatting, cleanup

Browse source
This commit is contained in:
Andreas Koepf 2025-01-24 17:12:42 +01:00
parent b767e58e48
commit 3dc80be7d2
12 changed files with 189 additions and 376 deletions
Download patch file Download diff file Expand all files Collapse all files

4
reasoning_gym/arithmetic/gcd.py
View file

@ -39,7 +39,7 @@ class GCDDataset(ProceduralDataset):
def _generate_numbers(self, rng: Random) -> Tuple[List[int], int]:
"""Generate a list of random positive integers and their GCD.
Will try up to 3 times to find numbers with GCD > 1."""
# Try up to 3 times to get GCD > 1
for _ in range(3):
num_count = rng.randint(self.config.min_numbers, self.config.max_numbers)
@ -47,7 +47,7 @@ class GCDDataset(ProceduralDataset):
result = reduce(gcd, numbers)
if result > 1:
break
# Return the last generated numbers, whether they met the criteria or not
return numbers, result