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

17
reasoning_gym/geometry/advanced_geometry.py
View file

@ -1,5 +1,5 @@
import random
from dataclasses import dataclass
from dataclasses import dataclass, field
from typing import List, Optional
import sympy
@ -21,16 +21,11 @@ class AdvancedGeometryConfig:
# Probability or list of tasks we want to generate
# For demonstration, we have three categories:
task_types: List[str] = None
def __post_init__(self):
if self.task_types is None:
# Default set of advanced tasks
self.task_types = [
"orthocenter",
"incircle_radius",
"angle_measure",
]
task_types: List[str] = field(default_factory=lambda: [
"orthocenter",
"incircle_radius",
"angle_measure",
])
def validate(self):
assert self.min_coord < self.max_coord, "min_coord must be < max_coord."