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

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)