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

add word ladder curriculum (#361)

Browse source 
* add word ladder curriculum

* add to __init__.py
This commit is contained in:
Oliver Stanley 2025-03-14 15:10:52 +00:00 • committed by GitHub
parent 8f8bd9d756
commit b5651e5e2c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 55 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

33
reasoning_gym/algorithmic/word_ladder.py
View file

@ -5,6 +5,7 @@ from dataclasses import dataclass
from random import Random
from typing import Any, Optional
from ..coaching import AttributeType, BaseCurriculum, RangeAttributeDefinition
from ..data import get_data_file_path
from ..factory import ProceduralDataset, register_dataset
@ -217,7 +218,16 @@ class WordLadderDataset(ProceduralDataset):
return {
"question": QUESTION_TEMPLATE.format(start=start, end=end),
"answer": ",".join(path),
"metadata": {"start_word": start, "end_word": end, "word_length": length, "chain_length": len(path)},
"metadata": {
"start_word": start,
"end_word": end,
"word_length": length,
"chain_length": len(path),
"difficulty": {
"word_length": length,
"chain_length": len(path),
},
},
}
def score_answer(self, answer: Optional[str], entry: dict[str, Any]) -> float:
@ -259,4 +269,23 @@ class WordLadderDataset(ProceduralDataset):
return reward
register_dataset("word_ladder", WordLadderDataset, WordLadderConfig)
class WordLadderCurriculum(BaseCurriculum):
def __init__(self):
super().__init__(WordLadderCurriculum.__name__, WordLadderConfig)
# Define attributes
self._define_attributes(
RangeAttributeDefinition(
name="word_length",
levels=[3, 4, 5, 6],
default_level=1,
description="Length of words in the puzzle",
attr_type=AttributeType.APPEND,
min_value=2,
lower_field_name="min_word_length",
upper_field_name="max_word_length",
)
)
register_dataset("word_ladder", WordLadderDataset, WordLadderConfig, WordLadderCurriculum)