o
    6ti                     @   s*   d dl Z dd ZdefddZdd ZdS )	    Nc                 C   s`   t | d }| d | d g}ddg}|| ||d  ||d  gt|ddkd} | S )Nactivitybaseliner      )	scenarioslabel)randomRandomshuffleintindex)docrndr   ordering r   X/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/hendrycks_ethics/utils.py_preproc_doc   s   
r   returnc                 C   s*   t | } d| d d  d| d d  dS )NzScenario 1: r   r   z
Scenario 2: r   z,
Question: Is Scenario 1 preferable?
Answer:r   r   r   r   r   doc_to_text   s   "r   c                 C   s   t | } | d S )Nr   r   r   r   r   r   doc_to_target   s   r   )r   r   strr   r   r   r   r   r   <module>   s    