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

Add allow-negation in products dataset

Browse source
This commit is contained in:
ahmedsaif 2025-02-20 12:01:09 +02:00
parent bd3ef0ee65
commit df4bb0b897
2 changed files with 36 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
reasoning_gym/arithmetic/products.py
View file

@ -14,6 +14,7 @@ class ProductsConfig:
max_terms: int = 2
min_digits: int = 1
max_digits: int = 5
allow_negation: bool = False
seed: Optional[int] = None
size: int = 500
@ -51,7 +52,10 @@ class ProductsDataset(ProceduralDataset):
num_digits = rng.randint(self.config.min_digits, self.config.max_digits)
# Calculate value ranges based on number of digits
min_value = 0 if num_digits == 1 else 10 ** (num_digits - 1) # Special case for 1 digit
if self.config.allow_negation:
min_value = -1 * 10 ** (num_digits) + 1
else:
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(rng, num_terms, min_value, max_value)