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 curriculum to ab dataset (#345)

Browse source 
* Add curriculum to ab dataset

* Add difficulty to metadata
This commit is contained in:
Adefioye 2025-03-13 15:03:02 -05:00 • committed by GitHub
parent 596de511f0
commit 4ec1154b47
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 66 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

3
reasoning_gym/algorithmic/__init__.py
View file

@ -6,7 +6,7 @@ Algorithmic tasks for training reasoning capabilities:
- Pattern matching
"""
from .ab import ABConfig, ABDataset
from .ab import ABConfig, ABCurriculum, ABDataset
from .base_conversion import BaseConversionConfig, BaseConversionCurriculum, BaseConversionDataset
from .binary_alternation import BinaryAlternationConfig, BinaryAlternationCurriculum, BinaryAlternationDataset
from .binary_matrix import BinaryMatrixConfig, BinaryMatrixCurriculum, BinaryMatrixDataset
@ -121,6 +121,7 @@ __all__ = [
"PoolMatrixCurriculum",
"ABConfig",
"ABDataset",
"ABCurriculum",
"CountPrimesConfig",
"CountPrimesDataset",
"CountPrimesCurriculum",