o
    6ti                     @   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| d  d	| d
  d| d  d| d  d}|g dt | d d d}|S )NuJ   다음을 읽고 정답으로 알맞은 것을 고르시요.
### Context: contextz
### Question: questionz
### Options:
(1) zoption#1z
(2) zoption#2z
(3) zoption#3z
(4) zoption#4z
(5) zoption#5u*   
### Answer: 주어진 문제의 정답은)z(1)z(2)z(3)z(4)z(5)gold   )r   choicesr   )int)docinstructionout_doc r   N/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/csatqa/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    