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

post merge formatting

Browse source
This commit is contained in:
Andreas Koepf 2025-02-02 15:24:39 +01:00
parent 5fe89932fe
commit 94eeff3255
4 changed files with 29 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

2
reasoning_gym/arithmetic/__init__.py
View file

@ -8,6 +8,7 @@ Arithmetic tasks for training reasoning capabilities:
"""
from .basic_arithmetic import BasicArithmeticDataset, BasicArithmeticDatasetConfig
from .calendar_arithmetic import CalendarArithmeticConfig, CalendarArithmeticDataset
from .chain_sum import ChainSum, ChainSumConfig
from .fraction_simplification import FractionSimplificationConfig, FractionSimplificationDataset
from .gcd import GCDConfig, GCDDataset
@ -15,7 +16,6 @@ from .lcm import LCMConfig, LCMDataset
from .leg_counting import LegCountingConfig, LegCountingDataset
from .prime_factorization import PrimeFactorizationConfig, PrimeFactorizationDataset
from .time_intervals import TimeIntervalsConfig, TimeIntervalsDataset
from .calendar_arithmetic import CalendarArithmeticConfig, CalendarArithmeticDataset
__all__ = [
"BasicArithmeticDataset",

8
reasoning_gym/arithmetic/calendar_arithmetic.py
View file

@ -1,10 +1,10 @@
import calendar
import math
import random
from dataclasses import dataclass
from datetime import date, timedelta
import calendar
import random
import math
from typing import Optional, Tuple, List, Dict, Any
from enum import Enum, auto
from typing import Any, Dict, List, Optional, Tuple
from ..factory import ProceduralDataset, register_dataset