File size: 408 Bytes
31dc8dc
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
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']