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

[aiw] remove output format template

Browse source
This commit is contained in:
rishabhranawat 2025-02-01 16:33:08 -08:00
parent f8696d6d22
commit 5f6d615369
1 changed files with 0 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

14
reasoning_gym/logic/aiw.py
View file

@ -113,20 +113,6 @@ class AliceInWonderlandDataset(ProceduralDataset):
],
}
self.format_templates = {
OutputFormat.PLAIN: Template("$question"),
OutputFormat.RESTRICTED: Template(
"$question To answer the question, DO NOT OUTPUT ANY TEXT EXCEPT "
'following format that contains final answer: "### Answer:"'
),
OutputFormat.THINKING: Template(
"$question Before providing answer to this problem, think "
"carefully step by step and double check the path to the "
'correct solution for any mistakes. Provide then the final '
'answer in following form: "### Answer:"'
),
}
def _get_aiw(self, rng: Random) -> dict:
"""Generates a single Alice in Wonderland question.