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

refactor: Rename WordReversalDataset to WordSequenceReversalDataset

Browse source
This commit is contained in:
Andreas Koepf (aider) 2025-01-26 11:52:15 +01:00
parent 4d582387de
commit c641b25508
4 changed files with 25 additions and 25 deletions
Download patch file Download diff file Expand all files Collapse all files

6
reasoning_gym/algorithmic/__init__.py
View file

@ -13,7 +13,7 @@ from .letter_jumble import LetterJumbleConfig, LetterJumbleDataset
from .number_filtering import NumberFilteringConfig, NumberFilteringDataset
from .number_sorting import NumberSortingConfig, NumberSortingDataset
from .spell_backward import SpellBackwardConfig, SpellBackwardDataset
from .word_reversal import WordReversalConfig, WordReversalDataset
from .word_reversal import WordSequenceReversalConfig, WordSequenceReversalDataset
__all__ = [
"SpellBackwardConfig",
@ -30,6 +30,6 @@ __all__ = [
"NumberFilteringDataset",
"NumberSortingConfig",
"NumberSortingDataset",
"WordReversalConfig",
"WordReversalDataset",
"WordSequenceReversalConfig",
"WordSequenceReversalDataset",
]

10
reasoning_gym/algorithmic/word_reversal.py
View file

@ -10,8 +10,8 @@ from ..factory import ProceduralDataset, register_dataset
@dataclass
class WordReversalConfig:
"""Configuration for word reversal task generation"""
class WordSequenceReversalConfig:
"""Configuration for word sequence reversal task generation"""
min_words: int = 3 # Minimum words in list
max_words: int = 8 # Maximum words in list
@ -24,8 +24,8 @@ class WordReversalConfig:
assert self.max_words >= self.min_words, "max_words must be >= min_words"
class WordReversalDataset(ProceduralDataset):
"""Generates word reversal tasks from text spans"""
class WordSequenceReversalDataset(ProceduralDataset):
"""Generates word sequence reversal tasks from text spans"""
def __init__(self, config: WordReversalConfig):
super().__init__(config=config, seed=config.seed, size=config.size)
@ -55,4 +55,4 @@ class WordReversalDataset(ProceduralDataset):
}
register_dataset("word_reversal", WordReversalDataset, WordReversalConfig)
register_dataset("word_sequence_reversal", WordSequenceReversalDataset, WordSequenceReversalConfig)