o
    7t¾ià  ã                   @   s"   d dl Z de jde jfdd„ZdS )é    NÚdatasetÚreturnc                 C   s   dd„ }|   |¡S )Nc                 S   sH   g d¢}t | d ƒ}|| }| d | d | d | d | d g|dœ}|S )	N)ÚAÚBÚCÚDÚanswerÚquestionÚchoice1Úchoice2Úchoice3Úchoice4)Ú	questionsÚchoicesr   )Úint)ÚdocÚanswer_listÚanswer_indexÚanswer_letterÚout_doc© r   ú\/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/mmlusr/question_only/utils.pyÚ_helper   s   ýzprocess_docs.<locals>._helper)Úmap)r   r   r   r   r   Úprocess_docs   s   
r   )ÚdatasetsÚDatasetr   r   r   r   r   Ú<module>   s    