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

fix(curriculum): Make boundaries in curriculum more sensible (#407)

Browse source 
* init

* fix tests

* unify codeio

* filtered for libraries not present in reasoning-gym

* fix more bounds

* puzzle24

* knight swap curriculum

* fix number sorting

* fix attributes

* add validation of config in creation of dataset

* dry run for instantiating and validating the datasets

* remove unused imports

* fix curriculum tests to reference newly updated attribute names
This commit is contained in:
Zafir Stojanovski 2025-04-04 20:24:14 +02:00 • committed by GitHub
parent 7853263650
commit dced3bfc45
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
132 changed files with 1226 additions and 347 deletions
Download patch file Download diff file Expand all files Collapse all files

7
reasoning_gym/arithmetic/gcd.py
View file

@ -71,7 +71,7 @@ class GCDDataset(ProceduralDataset):
"num_terms": num_terms,
"difficulty": {
"num_terms": (self.config.min_numbers, self.config.max_numbers),
"max_value": (self.config.min_value, self.config.max_value),
"value": (self.config.min_value, self.config.max_value),
},
},
}
@ -91,13 +91,14 @@ class GCDCurriculum(BaseCurriculum):
upper_field_name="max_numbers",
),
RangeAttributeDefinition(
name="max_value",
name="value",
levels=[100, 1000, 10000, 100000],
description="maximum value",
lower_field_name="min_value",
upper_field_name="max_value",
ensure_interval=True,
),
)
register_dataset(DATASET_NAME, GCDDataset, GCDConfig)
register_dataset(DATASET_NAME, GCDDataset, GCDConfig, GCDCurriculum)