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 score_answer method to ReseedingDataset

Browse source
This commit is contained in:
Andreas Koepf (aider) 2025-01-30 21:59:50 +01:00
parent 6d59648264
commit 66f99be4a3
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
reasoning_gym/dataset.py
View file

@ -118,3 +118,7 @@ class ReseedingDataset(Iterable[Dict[str, Any]]):
item = self._current_dataset[self._current_idx]
self._current_idx += 1
return item
def score_answer(self, answer: Optional[str], entry: Dict[str, any]) -> float:
"""Forward scoring to the wrapped dataset's implementation"""
return self.dataset.score_answer(answer, entry)