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: Update type hints and remove unused imports in bitwise_arithmetic.py

Browse source
This commit is contained in:
Andreas Koepf 2025-02-21 17:13:36 +01:00 • committed by Andreas Koepf (aider)
parent c91d13bd08
commit 44f4cc08eb
1 changed files with 3 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

7
reasoning_gym/arithmetic/bitwise_arithmetic.py
View file

@ -1,7 +1,6 @@
import ast
from dataclasses import dataclass
from random import Random
from typing import Any, Dict, List, Optional
from typing import Any, Optional
from ..factory import ProceduralDataset, register_dataset
@ -110,7 +109,7 @@ class BitwiseArithmeticDataset(ProceduralDataset):
def __init__(self, config: BitwiseArithmeticConfig) -> None:
super().__init__(config=config, seed=config.seed, size=config.size)
def __getitem__(self, idx: int) -> Dict[str, Any]:
def __getitem__(self, idx: int) -> dict[str, Any]:
"""
Generate a single arithmetic task.
@ -134,7 +133,7 @@ class BitwiseArithmeticDataset(ProceduralDataset):
return {"question": problem_str, "answer": answer, "metadata": {"problem": problem}}
def score_answer(self, answer: Optional[str], entry: Dict[str, Any]) -> float:
def score_answer(self, answer: Optional[str], entry: dict[str, Any]) -> float:
"""
Compares the user's answer with the correct answer.