open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-22 16:49:06 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: Add type hints to generate_puzzle and min_moves_n functions

Browse source
This commit is contained in:
Andreas Koepf (aider) 2025-02-20 22:38:11 +01:00
parent d7db2ed838
commit ece040a0d0
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
reasoning_gym/algorithmic/jugs.py
View file

@ -9,7 +9,7 @@ from typing import Dict, List, Optional, Tuple
from ..factory import ProceduralDataset, register_dataset
def min_moves_n(jug_capacities, target):
def min_moves_n(jug_capacities: List[int], target: int) -> Optional[int]:
"""
Compute the minimum number of moves required to have exactly `target` gallons
in any one jug for a puzzle with multiple jugs.
@ -73,7 +73,7 @@ def min_moves_n(jug_capacities, target):
return None
def generate_puzzle(rng, num_jugs=3, difficulty=6, max_attempts=10000):
def generate_puzzle(rng: Random, num_jugs: int = 3, difficulty: int = 6, max_attempts: int = 10000) -> Dict[str, any]:
"""
Generate a multi-jug water puzzle.