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 reasoning_gym.create_dataset({name}, ...) global factory function

Browse source
This commit is contained in:
Andreas Koepf 2025-01-25 00:58:34 +01:00
parent 0d2d8ba6a0
commit 519e411fa5
35 changed files with 133 additions and 598 deletions
Download patch file Download diff file Expand all files Collapse all files

17
reasoning_gym/algorithmic/word_reversal.py
View file

@ -6,7 +6,7 @@ from random import Random
from typing import List, Optional
from ..data import read_data_file
from ..dataset import ProceduralDataset
from ..factory import ProceduralDataset, register_dataset
@dataclass
@ -55,17 +55,4 @@ class WordReversalDataset(ProceduralDataset):
}
def word_reversal_dataset(
min_words: int = 3,
max_words: int = 8,
seed: Optional[int] = None,
size: int = 500,
) -> WordReversalDataset:
"""Create a WordReversalDataset with the given configuration."""
config = WordReversalConfig(
min_words=min_words,
max_words=max_words,
seed=seed,
size=size,
)
return WordReversalDataset(config)
register_dataset("word_reversal", WordReversalDataset, WordReversalConfig)