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

add to init

Browse source
This commit is contained in:
Rich Jones 2025-02-21 12:07:17 +01:00
parent 17088e9b42
commit ee9202d63d
2 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
reasoning_gym/arithmetic/__init__.py
View file

@ -3,6 +3,7 @@ Arithmetic tasks for training reasoning capabilities:
"""
from .basic_arithmetic import BasicArithmeticDataset, BasicArithmeticDatasetConfig
from .bitwise_arithmetic import BitwiseArithmeticConfig, BitwiseArithmeticDataset
from .calendar_arithmetic import CalendarArithmeticConfig, CalendarArithmeticDataset
from .chain_sum import ChainSumConfig, ChainSumDataset
from .count_bits import CountBitsConfig, CountBitsDataset
@ -55,4 +56,6 @@ __all__ = [
"DecimalArithmeticDataset",
"DecimalChainSumConfig",
"DecimalChainSumDataset",
"BitwiseArithmeticConfig",
"BitwiseArithmeticDataset",
]

2
reasoning_gym/arithmetic/bitwise_arithmetic.py
View file

@ -150,4 +150,4 @@ class BitwiseArithmeticDataset(ProceduralDataset):
# Register the dataset with the factory.
register_dataset("Bitwise_arithmetic", BitwiseArithmeticDataset, BitwiseArithmeticConfig)
register_dataset("bitwise_arithmetic", BitwiseArithmeticDataset, BitwiseArithmeticConfig)