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

Initial Principal and AttributeMonitor

Browse source
This commit is contained in:
EduardDurech 2025-02-05 11:17:31 +00:00
parent 3f5e7604c9
commit 45701be61e
6 changed files with 490 additions and 88 deletions
Download patch file Download diff file Expand all files Collapse all files

165
reasoning_gym/curricula/arithmetic/chain_sum_curriculum.py
View file

@ -1,93 +1,90 @@
"""
Curriculum definition for the ChainSum exercise.
This file defines the templates, attributes, and difficulty levels for generating chain sum problems.
"""
from typing import Dict, List, Union, Any
from typing import Dict, List, Any
from reasoning_gym.core.base_curriculum import BaseCurriculum, Template
from reasoning_gym.core.attributes import AttributeDefinition, AttributeType
# Define which attribute types are valid for this curriculum
ATTRIBUTE_TYPES = {
AttributeType.STATIC, # For base numbers
AttributeType.UBOUND, # For ranges like digits and terms
AttributeType.APPEND # For operators and notations
}
class ChainSumCurriculum(BaseCurriculum):
def __init__(self):
super().__init__("ChainSumCurriculum")
# Curriculum definition
CURRICULUM_NAME = "ChainSumCurriculum"
ATTRIBUTES = {
"num_digits": AttributeDefinition(
levels=[1, 2, 3, 4],
current_level=0, # Start with 1-digit numbers
description="Number of digits in each operand",
attr_type=AttributeType.UBOUND
),
"num_decimals": AttributeDefinition(
levels=[0, 1, 2],
current_level=0, # Start with integers
description="Number of decimal places in operands",
attr_type=AttributeType.UBOUND
),
"operators": AttributeDefinition(
levels=['+', '-', '*', '/', '**'],
current_level=0, # Start with basic operators
description="Set of operators that can be used, each level includes all previous operators",
attr_type=AttributeType.APPEND
),
"max_terms": AttributeDefinition(
levels=[2, 3, 4, 5],
current_level=0, # Start with 2 terms
description="Maximum number of terms in the expression",
attr_type=AttributeType.UBOUND
),
"sign": AttributeDefinition(
levels=['', '+', '-'],
current_level=0, # Start without negatives
description="Whether negative numbers are allowed",
attr_type=AttributeType.APPEND
),
"notation": AttributeDefinition(
levels=["regular", "scientific"],
current_level=0,
description="The notation to use for the expression",
attr_type=AttributeType.APPEND
),
"base": AttributeDefinition(
levels=[10, 2, 16],
current_level=0,
description="The base to use for the expression",
attr_type=AttributeType.STATIC
)
}
# Validate attributes use allowed types (and include the curriculum name in error messages)
AttributeDefinition.validate_attributes(ATTRIBUTES, ATTRIBUTE_TYPES, curriculum=CURRICULUM_NAME)
# Template definitions
TEMPLATES = [
{
"question": "What is {expression}?",
"answer": "{result}",
"metadata": {"type": "direct"}
},
{
"question": "Calculate the following: {expression}",
"answer": "{result}",
"metadata": {"type": "direct"}
},
{
"question": "Solve {expression}",
"answer": "{result}",
"metadata": {"type": "direct"}
}
]
def _init_curriculum(self) -> None:
"""Initialize the ChainSum curriculum configuration"""
# Define valid attribute types
self._valid_types = {
AttributeType.STATIC, # For base numbers
AttributeType.UBOUND, # For ranges like digits and terms
AttributeType.APPEND # For operators and notations
}
# Define attributes
self._attributes = {
"num_digits": AttributeDefinition(
levels=[2, 4, 10],
default_level=0, # Start with 1-digit numbers
description="Number of digits in each operand",
attr_type=AttributeType.UBOUND
# distribution_type
),
"num_decimals": AttributeDefinition(
levels=[0, 1, 2],
default_level=0, # Start with integers
description="Number of decimal places in operands",
attr_type=AttributeType.UBOUND
),
"operators": AttributeDefinition(
levels=['+', '-', '*', '/', '**'],
default_level=0, # Start with basic operators
description="Set of operators that can be used",
attr_type=AttributeType.APPEND
),
"max_terms": AttributeDefinition(
levels=[2, 3, 4, 5],
default_level=0, # Start with 2 terms
description="Maximum number of terms in the expression",
attr_type=AttributeType.UBOUND
),
"sign": AttributeDefinition(
levels=['', '+', '-'],
default_level=0, # Start without negatives
description="Whether negative numbers are allowed",
attr_type=AttributeType.APPEND
),
"notation": AttributeDefinition(
levels=["regular", "scientific"],
default_level=0,
description="The notation to use for the expression",
attr_type=AttributeType.APPEND
),
"base": AttributeDefinition(
levels=[10, 2, 16],
default_level=0,
description="The base to use for the expression",
attr_type=AttributeType.STATIC
)
}
# Define templates
self._templates = [
Template(
question="What is {expression}?",
answer="{result}",
metadata={"type": "direct"}
),
Template(
question="Calculate the following: {expression}",
answer="{result}",
metadata={"type": "direct"}
),
Template(
question="Solve {expression}",
answer="{result}",
metadata={"type": "direct"}
)
]
# Generator functions for placeholders
def generate_expression(attributes: Dict[str, Any]) -> Dict[str, str]:
@ -102,4 +99,4 @@ def generate_expression(attributes: Dict[str, Any]) -> Dict[str, str]:
Dict containing the expression and result as strings
"""
# This will be implemented in the Exercise class
pass
pass