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: Update fraction simplification config to use Sequence and default tuple for styles

Browse source
This commit is contained in:
Andreas Koepf (aider) 2025-01-24 09:42:07 +01:00
parent 9826d98fcf
commit e64e82c9e2
1 changed files with 3 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

10
reasoning_gym/arithmetic/fraction_simplification.py
View file

@ -1,7 +1,7 @@
"""Fraction simplification task generator"""
from dataclasses import dataclass
from random import Random
from typing import List, Optional, Tuple
from typing import Optional, Tuple, Sequence
from math import gcd
@ -12,7 +12,7 @@ class FractionSimplificationConfig:
max_value: int = 100 # Maximum value for numerator/denominator
min_factor: int = 2 # Minimum multiplication factor
max_factor: int = 10 # Maximum multiplication factor
styles: List[str] = None # List of allowed fraction formatting styles
styles: Sequence[str] = ("plain", "latex_inline", "latex_frac", "latex_dfrac") # Allowed fraction formatting styles
seed: Optional[int] = None
size: int = 500 # Virtual dataset size
@ -23,10 +23,6 @@ class FractionSimplificationConfig:
assert self.min_factor >= 2, "min_factor must be at least 2"
assert self.max_factor >= self.min_factor, "max_factor must be >= min_factor"
# Set default styles if none provided
if self.styles is None:
self.styles = ["plain", "latex_inline", "latex_frac", "latex_dfrac"]
# Validate styles
valid_styles = {"plain", "latex_inline", "latex_frac", "latex_dfrac"}
for style in self.styles:
@ -140,7 +136,7 @@ def fraction_simplification_dataset(
max_value: int = 100,
min_factor: int = 2,
max_factor: int = 10,
styles: List[str] = None,
styles: Sequence[str] = ("plain", "latex_inline", "latex_frac", "latex_dfrac"),
seed: Optional[int] = None,
size: int = 500,
) -> FractionSimplificationDataset: