| import os | |
| from collections import defaultdict | |
| from typing import Any | |
| import io | |
| import numpy as np | |
| from tqdm import tqdm | |
| QUERY_TEMPLATE_MATH = """ | |
| Please answer the question step by step and put the answer in \\boxed{{}}. {question} | |
| """.strip() | |
| def format_prompt(row: dict): | |
| prompt_messages = QUERY_TEMPLATE_MATH.format( | |
| question=row['problem'] | |
| ) | |
| return prompt_messages, row['answer'] | |