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

feat: Add GCD dataset imports to arithmetic module

Browse source
This commit is contained in:
Andreas Koepf (aider) 2025-01-24 08:51:29 +01:00
parent 6cf01debdb
commit d7ef1d471b
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
reasoning_gym/arithmetic/__init__.py
View file

@ -8,6 +8,7 @@ Arithmetic tasks for training reasoning capabilities:
from .basic_arithmetic import ArithmeticDataset, ArithmeticDatasetConfig, arithmetic_dataset from .basic_arithmetic import ArithmeticDataset, ArithmeticDatasetConfig, arithmetic_dataset
from .chain_sum import ChainSum, ChainSumConfig, chain_sum_dataset from .chain_sum import ChainSum, ChainSumConfig, chain_sum_dataset
from .gcd import GCDConfig, GCDDataset, gcd_dataset
from .leg_counting import LegCountingConfig, LegCountingDataset, leg_counting_dataset from .leg_counting import LegCountingConfig, LegCountingDataset, leg_counting_dataset
from .prime_factorization import PrimeFactorizationConfig, PrimeFactorizationDataset, prime_factorization_dataset from .prime_factorization import PrimeFactorizationConfig, PrimeFactorizationDataset, prime_factorization_dataset
@ -18,6 +19,9 @@ __all__ = [
"ChainSum", "ChainSum",
"ChainSumConfig", "ChainSumConfig",
"chain_sum_dataset", "chain_sum_dataset",
"GCDConfig",
"GCDDataset",
"gcd_dataset",
"LegCountingConfig", "LegCountingConfig",
"LegCountingDataset", "LegCountingDataset",
"leg_counting_dataset", "leg_counting_dataset",