o
    6t¾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   sL   | d   ¡ d | d   ¡  d | d   ¡  }| d | d g}||ddœ}|S )	NÚnormú Ú	situationÚ	intentionÚmoral_actionÚimmoral_actionr   )ÚqueryÚchoicesÚlabel)Ú
capitalize)ÚdocÚctxr   Úout_doc© r   úY/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/histoires_morales/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    