open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-27 17:23:19 +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 3a8a613573
commit a20a9e9650
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 self._current_idx = 0
return self return self
def __next__(self): def __next__(self) -> Dict[str, Any]:
"""Get next item in iteration""" """Get next item in iteration"""
if self._current_idx >= self.size: if self._current_idx >= self.size:
raise StopIteration raise StopIteration