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

Curriculum/cognition (#314)

Browse source 
* added rectangle count curriculum

* added number sequences

* registered curriculum
This commit is contained in:
joesharratt1229 2025-03-11 00:10:28 +01:00 • committed by GitHub
parent d0b49cfffd
commit 0dce7adbad
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 117 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

20
reasoning_gym/cognition/number_sequences.py
View file

@ -3,6 +3,7 @@ from enum import StrEnum
from random import Random
from typing import Optional
from ..coaching import AttributeType, BaseCurriculum, ScalarAttributeDefinition
from ..factory import ProceduralDataset, register_dataset
@ -198,4 +199,21 @@ class NumberSequenceDataset(ProceduralDataset):
}
register_dataset("number_sequence", NumberSequenceDataset, NumberSequenceConfig)
class NumberSequenceCurriculum(BaseCurriculum):
def __init__(self):
super().__init__(NumberSequenceCurriculum.__name__, NumberSequenceConfig)
self._define_attributes(
ScalarAttributeDefinition(
name="max_complexity",
levels=[1, 2, 3, 4],
default_level=0,
description="Maximum number of operations to combine",
attr_type=AttributeType.STATIC,
min_value=1,
field_name="max_complexity",
),
)
register_dataset("number_sequence", NumberSequenceDataset, NumberSequenceConfig, NumberSequenceCurriculum)