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

added basic arith curricula (#276)

Browse source 
* added basic arith curricula
* register BasicArithmeticCurriculum

---------

Co-authored-by: Andreas Koepf <andreas.koepf@provisio.com>
This commit is contained in:
joesharratt1229 2025-03-07 22:54:49 +01:00 • committed by GitHub
parent f490b9f760
commit 1888fe2bb4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 82 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

3
reasoning_gym/arithmetic/__init__.py
View file

@ -2,7 +2,7 @@
Arithmetic tasks for training reasoning capabilities:
"""
from .basic_arithmetic import BasicArithmeticDataset, BasicArithmeticDatasetConfig
from .basic_arithmetic import BasicArithmeticCurriculum, BasicArithmeticDataset, BasicArithmeticDatasetConfig
from .bitwise_arithmetic import BitwiseArithmeticConfig, BitwiseArithmeticDataset
from .calendar_arithmetic import CalendarArithmeticConfig, CalendarArithmeticDataset
from .chain_sum import ChainSumConfig, ChainSumDataset
@ -24,6 +24,7 @@ from .time_intervals import TimeIntervalsConfig, TimeIntervalsDataset
__all__ = [
"BasicArithmeticDataset",
"BasicArithmeticDatasetConfig",
"BasicArithmeticCurriculum",
"ChainSumDataset",
"ChainSumConfig",
"CalendarArithmeticConfig",