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

feat: Register chain_sum dataset with register_dataset function

Browse source
This commit is contained in:
Andreas Koepf (aider) 2025-01-25 00:01:41 +01:00
parent 2befe97151
commit df2b8d2809
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
reasoning_gym/arithmetic/chain_sum.py
View file

@ -3,6 +3,7 @@ from dataclasses import dataclass
from typing import Optional from typing import Optional
from ..dataset import ProceduralDataset from ..dataset import ProceduralDataset
from ..factory import register_dataset
@dataclass @dataclass
@ -143,3 +144,6 @@ def chain_sum_dataset(
size=size, size=size,
) )
return ChainSum(config) return ChainSum(config)
# Register the dataset
register_dataset("chain_sum", ChainSum, ChainSumConfig)