o
    6ti>                     @   s   d dl Z dd Zdd ZdS )    Nc                 C   s   t | d }|S )Nchoices)astliteral_eval)docr    r   P/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/afrimmlu/utils.pydoc_to_choice   s   r   c                 C   sD   d}t | d }|j| d | d |d |d |d |d d	}|S )
Na  You are a highly knowledgeable and intelligent artificial intelligence
                model answers multiple-choice questions about '{subject}'

                Question: '''{question}'''

                Choices:
                        A: ''{choice1}'''
                        B: ''{choice2}'''
                        C: ''{choice3}'''
                        D: ''{choice4}'''

                Answer:  r   subjectquestionr            )r	   r
   choice1choice2choice3choice4)r   r   format)r   outputr   textr   r   r   doc_to_text	   s   r   )r   r   r   r   r   r   r   <module>   s    