From 5f6d6153696c3b213ef287e6f4e0a226d3439480 Mon Sep 17 00:00:00 2001 From: rishabhranawat Date: Sat, 1 Feb 2025 16:33:08 -0800 Subject: [PATCH] [aiw] remove output format template --- reasoning_gym/logic/aiw.py | 14 -------------- 1 file changed, 14 deletions(-) diff --git a/reasoning_gym/logic/aiw.py b/reasoning_gym/logic/aiw.py index ebeb50a7..7a27b6d0 100644 --- a/reasoning_gym/logic/aiw.py +++ b/reasoning_gym/logic/aiw.py @@ -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.