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

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/letter_counting.py
View file

@ -7,7 +7,7 @@ from typing import List, Optional
from reasoning_gym.data import read_data_file
from ..dataset import ProceduralDataset
from ..factory import ProceduralDataset, register_dataset
@dataclass
@ -63,17 +63,4 @@ class LetterCountingDataset(ProceduralDataset):
}
def letter_counting_dataset(
min_words: int = 5,
max_words: int = 15,
seed: Optional[int] = None,
size: int = 500,
) -> LetterCountingDataset:
"""Create a LetterCountingDataset with the given configuration."""
config = LetterCountingConfig(
min_words=min_words,
max_words=max_words,
seed=seed,
size=size,
)
return LetterCountingDataset(config)
register_dataset("letter_counting", LetterCountingDataset, LetterCountingConfig)