o
    7tij                     @   s    d e fddZd efddZdS )returnc                 C   sL   | d | d | d | d d}d dd | D }d	| d
  d| dS )Nending0ending1ending2ending3)ABCD c                 s   s$    | ]\}}| d | dV  qdS )z. 
N ).0kvr   r   X/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/medqa/preprocess_medqa.py	<genexpr>   s   " zdoc_to_text.<locals>.<genexpr>z
Question: sent1r   zAnswer:)joinitems)docoption_choicesanswersr   r   r   doc_to_text   s   r   c                 C   s   | d S )Nlabelr   )r   r   r   r   doc_to_target   s   r   N)strr   intr   r   r   r   r   <module>   s    