open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-28 17:29:39 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: Add leg counting imports and exports to arithmetic package

Browse source
This commit is contained in:
Andreas Koepf (aider) 2025-01-23 22:22:12 +01:00
parent f8219a7a6a
commit b8c69aa89f
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
reasoning_gym/arithmetic/__init__.py
View file

@ -1,9 +1,13 @@
from .chain_sum import ChainSum, ChainSumConfig from .chain_sum import ChainSum, ChainSumConfig
from .basic_arithmetic import ArithmeticDataset, ArithmeticDatasetConfig from .basic_arithmetic import ArithmeticDataset, ArithmeticDatasetConfig
from .leg_counting import LegCountingConfig, LegCountingDataset, leg_counting_dataset
__all__ = [ __all__ = [
"ChainSum", "ChainSum",
"ChainSumConfig", "ChainSumConfig",
"ArithmeticDataset", "ArithmeticDataset",
"ArithmeticDatasetConfig" "ArithmeticDatasetConfig",
"LegCountingConfig",
"LegCountingDataset",
"leg_counting_dataset"
] ]