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

refactor: Use field default_factory TimeIntervalsConfig, AdvancedGeometryConfig

Browse source
This commit is contained in:
Andreas Koepf (aider) 2025-02-02 09:41:23 +01:00 • committed by Andreas Koepf
parent 8202f234be
commit 4e9fc4baad
3 changed files with 29 additions and 124 deletions
Download patch file Download diff file Expand all files Collapse all files

10
reasoning_gym/arithmetic/time_intervals.py
View file

@ -1,5 +1,5 @@
import random
from dataclasses import dataclass
from dataclasses import dataclass, field
from datetime import date, datetime, time, timedelta
from typing import List, Optional
@ -19,14 +19,12 @@ class TimeIntervalsConfig:
min_date: date = date(1900, 1, 1)
max_date: date = date(3000, 1, 1)
max_date_difference_days: int = 100
task_types: List[str] = None
task_types: List[str] = field(
default_factory=lambda: ["time", "time_seconds", "time_ms", "date", "datetime", "datetime_tz"]
)
seed: Optional[int] = None
size: int = 500
def __post_init__(self):
if self.task_types is None:
self.task_types = ["time", "time_seconds", "time_ms", "date", "datetime", "datetime_tz"]
def validate(self) -> None:
"""Validate configuration parameters"""
assert self.size > 0, "size must be positive"