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

fix: Move EpochTrackingDataLoader after ReasoningGymDataset to resolve undefined name error

Browse source
This commit is contained in:
Andreas Koepf (aider) 2025-02-22 21:12:15 +00:00 • committed by Andreas Koepf
parent 5f16d54ebe
commit 8dc6cb5228
4 changed files with 24 additions and 30 deletions
Download patch file Download diff file Expand all files Collapse all files

4
reasoning_gym/arithmetic/chain_sum.py
View file

@ -122,7 +122,7 @@ class ChainSumCurriculum(BaseCurriculum):
self._define_attributes(
RangeAttributeDefinition(
name="num_terms",
levels=[2, 3, 4, 5],
levels=list(range(2, 8)),
default_level=0, # Start with 2 terms
description="Maximum number of terms in the expression",
attr_type=AttributeType.APPEND,
@ -132,7 +132,7 @@ class ChainSumCurriculum(BaseCurriculum):
),
RangeAttributeDefinition(
name="num_digits",
levels=[1, 2, 4, 10],
levels=list(range(1, 10)),
default_level=0, # Start with 1-digit numbers
description="Number of digits in each operand",
attr_type=AttributeType.APPEND,

4
reasoning_gym/arithmetic/products.py
View file

@ -114,7 +114,7 @@ class ProductsCurriculum(BaseCurriculum):
self._define_attributes(
RangeAttributeDefinition(
name="num_terms",
levels=[2, 3, 4, 5],
levels=list(range(2, 8)),
default_level=0, # Start with 2 terms
description="Maximum number of terms in the expression",
attr_type=AttributeType.APPEND,
@ -124,7 +124,7 @@ class ProductsCurriculum(BaseCurriculum):
),
RangeAttributeDefinition(
name="num_digits",
levels=[1, 2, 3, 4],
levels=list(range(1, 10)),
default_level=0, # Start with 1-digit numbers
description="Number of digits in each operand",
attr_type=AttributeType.APPEND,