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: Replace if/elif chain with match/case in _format_fraction

Browse source
This commit is contained in:
Andreas Koepf (aider) 2025-01-24 09:43:13 +01:00
parent 0cb01efbc5
commit 5fb6421179
1 changed files with 11 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

21
reasoning_gym/arithmetic/fraction_simplification.py
View file

@ -93,16 +93,17 @@ class FractionSimplificationDataset:
def _format_fraction(self, num: int, den: int, style: str = "plain") -> str:
"""Format a fraction in various styles"""
if style == "plain":
return f"{num}/{den}"
elif style == "latex_inline":
return f"${num}/{den}$"
elif style == "latex_frac":
return f"$\\frac{{{num}}}{{{den}}}$"
elif style == "latex_dfrac":
return f"$\\dfrac{{{num}}}{{{den}}}$"
else:
raise ValueError(f"Unknown fraction style: {style}")
match style:
case "plain":
return f"{num}/{den}"
case "latex_inline":
return f"${num}/{den}$"
case "latex_frac":
return f"$\\frac{{{num}}}{{{den}}}$"
case "latex_dfrac":
return f"$\\dfrac{{{num}}}{{{den}}}$"
case _:
raise ValueError(f"Unknown fraction style: {style}")
def __getitem__(self, idx: int) -> dict:
"""Generate a single fraction simplification task"""