o
    6ti
                     @   s   d dl mZ d dlmZ defddZdefddZdee fdd	Zd
edefddZd
edefddZ	d
edefddZ
dS )    )List)Datasetreturnc                 C   s   | d }| d }| d }|r-d| d| d|d  d|d	  d
|d  d|d  d}|S d| d|d  d|d	  d
|d  d|d  d}|S )N	paragraphquestionchoicesu   주어진 맥락을 천천히 읽고, 질문에 대한 적절한 정답을 A, B, C, D 중에 골라 알파벳 하나로 답하시오.

맥락: u	   
질문: u   
보기:
A:r   z, B:    z, C:    z, D:    u   
정답:u~   주어진 질문을 천천히 읽고, 적절한 정답을 A, B, C, D 중에 골라 알파벳 하나로 답하시오.

질문:  )docctxqoptresr   r   X/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/click/click_lang/utils.pyget_context   s   :4r   c                 C   s@   | d }d| d v rg d| d  | S g d| d  | S )NanswerCSATidABCDEr   r   r   r   r   )index)r   ansr   r   r   
get_target   s   r   c                 C   s   d| d v r
g dS g dS )Nr   r   r   r   r   )r   r   r   r   get_choices   s   r    datasetc                 C      |  dd S )Nc                 S   sX   d| d v p+d| d v ot | d dd dk p+d| d v o+t | d dd dkS )	NCSAT_korean_22r   CSAT_korean_23_#   TK   intsplitexampler   r   r   <lambda>!   s    $&zextract_text.<locals>.<lambda>filterr!   r   r   r   extract_text      r3   c                 C   r"   )Nc                 S   s   d| d v ot | d dd dk ot | d dd dkpVd| d v oC| d dd	 d
kpCd| d v pBd| d v pBd| d v  pVd| d v oVt | d dd dk S )NCSAT_koreanr   r%   r	      r
   
   Kedu_1r   16   대화r      발화   질의r(   r&      r*   r-   r   r   r   r/   +   s   

&z!extract_grammar.<locals>.<lambda>r0   r2   r   r   r   extract_grammar)   r4   r>   c                 C   r"   )Nc                 S   s   d| d v o,t | d dd dkp,t | d dd dk o,t | d dd d	k pJd
| d v oDd| d v pDd| d v pDd| d v pJd| d v S )Nr5   r   r%   r&   "   r	   r6   r
      Kedu_16r:   r   r;   r<   
PSE_koreanr*   r-   r   r   r   r/   C   s   

z"extract_function.<locals>.<lambda>r0   r2   r   r   r   extract_functionA   r4   rC   N)typingr   datasetsr   strr   r   r    r3   r>   rC   r   r   r   r   <module>   s    
