o
    6t¾ii  ã                   @   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   s0   | d }| d d | | d t | d ƒdœ}|S )NÚctxÚactivity_labelz: ÚendingsÚlabel)ÚqueryÚchoicesÚgold)Úint)Údocr   Úout_doc© r   úT/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/egyhellaswag/utils.pyÚ_process_doc   s   
ýz"process_docs.<locals>._process_doc)Úmap)r   r   r   r   r   Úprocess_docs   s   
	r   )ÚdatasetsÚDatasetr   r   r   r   r   Ú<module>   s    