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

base conversion curriculum (#286)

Browse source
This commit is contained in:
Zafir Stojanovski 2025-03-08 01:46:32 +01:00 • committed by GitHub
parent 7199363339
commit adf8cd8f6d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 67 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

3
reasoning_gym/algorithmic/__init__.py
View file

@ -7,7 +7,7 @@ Algorithmic tasks for training reasoning capabilities:
"""
from .ab import ABConfig, ABDataset
from .base_conversion import BaseConversionConfig, BaseConversionDataset
from .base_conversion import BaseConversionConfig, BaseConversionCurriculum, BaseConversionDataset
from .binary_alternation import BinaryAlternationConfig, BinaryAlternationCurriculum, BinaryAlternationDataset
from .binary_matrix import BinaryMatrixConfig, BinaryMatrixCurriculum, BinaryMatrixDataset
from .caesar_cipher import CaesarCipherConfig, CaesarCipherDataset
@ -46,6 +46,7 @@ __all__ = [
"SpellBackwardDataset",
"BaseConversionConfig",
"BaseConversionDataset",
"BaseConversionCurriculum",
"CaesarCipherConfig",
"CaesarCipherDataset",
"CryptarithmConfig",

36
reasoning_gym/algorithmic/base_conversion.py
View file

@ -4,6 +4,7 @@ from dataclasses import dataclass
from random import Random
from typing import Optional
from ..coaching import AttributeType, BaseCurriculum, RangeAttributeDefinition
from ..factory import ProceduralDataset, register_dataset
QUESTION_TEMPLATE = """Your task is to convert a number between two different bases.
@ -108,8 +109,41 @@ class BaseConversionDataset(ProceduralDataset):
"target_base": target_base,
"source_repr": source_repr,
"target_repr": target_repr,
"difficulty": {
"value": value,
"base": (source_base, target_base),
},
},
}
register_dataset("base_conversion", BaseConversionDataset, BaseConversionConfig)
class BaseConversionCurriculum(BaseCurriculum):
def __init__(self):
super().__init__(BaseConversionCurriculum.__name__, BaseConversionConfig)
# Define attributes
self._define_attributes(
RangeAttributeDefinition(
name="base",
levels=[9, 18, 27, 36],
default_level=0,
description="The base of the number system",
attr_type=AttributeType.APPEND,
min_value=2,
lower_field_name="min_base",
upper_field_name="max_base",
),
RangeAttributeDefinition(
name="value",
levels=[1_000, 10_000, 100_000, 1_000_000],
default_level=0,
description="The value to convert",
attr_type=AttributeType.APPEND,
min_value=0,
lower_field_name="min_value",
upper_field_name="max_value",
),
)
register_dataset("base_conversion", BaseConversionDataset, BaseConversionConfig, BaseConversionCurriculum)