feat: Expose arithmetic modules automatically in reasoning_gym package

This commit is contained in:
Andreas Koepf (aider) 2025-01-23 12:25:26 +01:00
parent 5c5cbf3194
commit 575624aa93
2 changed files with 10 additions and 1 deletions

View file

@ -1,3 +1,9 @@
from .chain_sum import ChainSum, ChainSumConfig
from .basic_arithmetic import ArithmeticDataset, ArithmeticDatasetConfig
__all__ = ["ChainSum", "ChainSumConfig"]
__all__ = [
"ChainSum",
"ChainSumConfig",
"ArithmeticDataset",
"ArithmeticDatasetConfig"
]