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: Add return type annotation to ProceduralDataset.__next__()

Browse source
This commit is contained in:
Andreas Koepf (aider) 2025-01-24 10:16:27 +01:00
parent 2a6a9655d7
commit 719d760eea
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
reasoning_gym/dataset.py
View file

@ -22,7 +22,7 @@ class ProceduralDataset(ABC, Sized, Iterable[Dict[str, Any]]):
self._current_idx = 0
return self
def __next__(self):
def __next__(self) -> Dict[str, Any]:
"""Get next item in iteration"""
if self._current_idx >= self.size:
raise StopIteration