open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-30 17:40:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(envs): Add source dataset and index to metadata (#388)

Browse source 
* add source dataset and index to metadata

* fix typo

* fix coach class and its test
This commit is contained in:
Zafir Stojanovski 2025-03-20 12:12:14 +01:00 • committed by GitHub
parent c6d01541aa
commit 4c47527130
104 changed files with 549 additions and 146 deletions
Download patch file Download diff file Expand all files Collapse all files

6
reasoning_gym/arithmetic/gsm_symbolic/gsm_symbolic.py
View file

@ -7,6 +7,8 @@ from typing import Any, Callable, Optional
from reasoning_gym.factory import ProceduralDataset, register_dataset
DATASET_NAME = "gsm_symbolic"
tasks_ok = [
0,
1,
@ -151,6 +153,8 @@ class GSMSymbolicDataset(ProceduralDataset):
generator = self.generators[generator_idx]
example = generator(rng, self.config.difficulty)
example["question"] += " Give the result as your final answer. Do not include units."
example["metadata"]["source_dataset"] = DATASET_NAME
example["metadata"]["source_index"] = idx
return example
def score_answer(self, answer: Optional[str], entry: dict[str, Any]) -> float:
@ -174,4 +178,4 @@ class GSMSymbolicDataset(ProceduralDataset):
return reward
register_dataset("gsm_symbolic", GSMSymbolicDataset, GSMSymbolicDatasetConfig)
register_dataset(DATASET_NAME, GSMSymbolicDataset, GSMSymbolicDatasetConfig)