open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-23 16:55:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: Add special case handling for min_digits=1 in ChainSum generation

Browse source
This commit is contained in:
Andreas Koepf (aider) 2025-01-23 12:07:56 +01:00
parent 4777e6b435
commit 516d4d20d4
2 changed files with 35 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
reasoning_gym/arithmetic/chain_sum.py
View file

@ -57,7 +57,7 @@ class ChainSum:
num_digits = item_rng.randint(self.config.min_digits, self.config.max_digits)
# Calculate value ranges based on number of digits
min_value = 10 ** (num_digits - 1) # e.g., 100 for 3 digits
min_value = 0 if num_digits == 1 else 10 ** (num_digits - 1) # Special case for 1 digit
max_value = (10 ** num_digits) - 1 # e.g., 999 for 3 digits
expression, result = self._generate_task(item_rng, num_terms, min_value, max_value)