open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-19 12:58:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: Add prime factorization task generator with configurable range and example

Browse source
This commit is contained in:
Andreas Koepf (aider) 2025-01-23 22:46:58 +01:00
parent 466b78a816
commit a4391fe5f6
2 changed files with 99 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
reasoning_gym/arithmetic/__init__.py
View file

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