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

Caesar cipher curriculum (#302)

Browse source 
* caesar cipher curriculum + tests
This commit is contained in:
vncntt 2025-03-08 23:23:32 -08:00 • committed by GitHub
parent e0f8ef061d
commit fc908d4cf4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 83 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

37
reasoning_gym/algorithmic/caesar_cipher.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 ..data import read_data_file
from ..factory import ProceduralDataset, register_dataset
@ -74,8 +75,40 @@ class CaesarCipherDataset(ProceduralDataset):
return {
"question": f"Decrypt this Caesar cipher text: {cipher_text}. Provide only the decrypted text as your final answer.",
"answer": sentence,
"metadata": {"rotation": rotation, "cipher_text": cipher_text, "clear_text": sentence},
"metadata": {
"rotation": rotation,
"cipher_text": cipher_text,
"clear_text": sentence,
},
}
register_dataset("caesar_cipher", CaesarCipherDataset, CaesarCipherConfig)
class CaesarCipherCurriculum(BaseCurriculum):
"""Curriculum for Caesar cipher task generation"""
def __init__(self):
super().__init__(CaesarCipherCurriculum.__name__, CaesarCipherConfig)
self._define_attributes(
RangeAttributeDefinition(
name="rotation",
levels=[5, 10, 15, 25],
default_level=0,
description="Max rotation for cipher",
attr_type=AttributeType.APPEND,
lower_field_name="min_rotation",
upper_field_name="max_rotation",
),
RangeAttributeDefinition(
name="words",
levels=[5, 10, 15, 25],
default_level=0,
description="Max number of words",
attr_type=AttributeType.APPEND,
lower_field_name="min_words",
upper_field_name="max_words",
),
)
register_dataset("caesar_cipher", CaesarCipherDataset, CaesarCipherConfig, CaesarCipherCurriculum)