feat(env): Binary Matrix Curriculum (#279)

* binary matrix curriculum

* register BinaryMatrixCurriculum

---------

Co-authored-by: Andreas Koepf <andreas.koepf@provisio.com>
This commit is contained in:
Zafir Stojanovski 2025-03-07 22:58:47 +01:00 committed by GitHub
parent 1888fe2bb4
commit 25b8e35589
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 68 additions and 7 deletions

View file

@ -9,7 +9,7 @@ Algorithmic tasks for training reasoning capabilities:
from .ab import ABConfig, ABDataset
from .base_conversion import BaseConversionConfig, BaseConversionDataset
from .binary_alternation import BinaryAlternationConfig, BinaryAlternationCurriculum, BinaryAlternationDataset
from .binary_matrix import BinaryMatrixConfig, BinaryMatrixDataset
from .binary_matrix import BinaryMatrixConfig, BinaryMatrixCurriculum, BinaryMatrixDataset
from .caesar_cipher import CaesarCipherConfig, CaesarCipherDataset
from .count_primes import CountPrimesConfig, CountPrimesDataset
from .cryptarithm import CryptarithmConfig, CryptarithmDataset
@ -89,6 +89,7 @@ __all__ = [
"ManipulateMatrixDataset",
"BinaryMatrixConfig",
"BinaryMatrixDataset",
"BinaryMatrixCurriculum",
"PoolMatrixConfig",
"PoolMatrixDataset",
"ABConfig",