o
    7tih                     @   s    d e fddZd efddZdS )returnc                 C   sh   g d}d| d  d }|d| d  d 7 }t || d D ]\}}||  d	| d7 }q|d
7 }|S )z
    Passage: <passage>
    Question: <question>
    Choices:
    A. <choice1>
    B. <choice2>
    C. <choice3>
    D. <choice4>
    Answer:
    abcdz	Passage: context
z
Question: questionz

Choices:
optionsz. zAnswer:)zipupper)docchoicespromptchoiceoption r   U/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/logiqa/utils_logiqa.pydoc_to_text   s   r   c                 C   s   g d}| | d  S )Nr   label)indexstrip)r   r   r   r   r   doc_to_target   s   r   N)strr   intr   r   r   r   r   <module>   s   