o
    6t¾iÿ  ã                   @   s&   d ddœZ dd„ Zdd„ Zdd„ Zd	S )
z
 Izan ere,z Beraz,)ÚcauseÚeffectc                 C   s   t | d  }| d  ¡ |›  S )NÚquestionÚpremise)Úxcopa_connectorsÚstrip)ÚdocÚconn© r	   úT/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/basque_bench/utils.pyÚxcopa_doc_to_text   s   r   c                 C   s    dd„ }|| d ƒ|| d ƒgS )Nc                 S   ó   | d   ¡ | dd …  S ©Né    é   ©Úlower)Úchoicer	   r	   r
   Úconvert_choice   ó   z+xcopa_doc_to_choice.<locals>.convert_choiceÚchoice1Úchoice2r	   )r   r   r	   r	   r
   Úxcopa_doc_to_choice   s   r   c                    s.   g ‰ ‡ ‡fdd„}dd„ ‰|   dd„ ¡ |¡S )Nc                    s\   | d dvr'| d dvr'| d   d¡r| d d d… | d< ˆ| d ƒ| d< | S ˆ  | ¡ | S )NÚ	sentence1©NÚ Ú	sentence2)Ú.ú,ú;éÿÿÿÿ)ÚendswithÚappend©r   ©Ú
empty_docsÚlowercase_first_letterr	   r
   Ú_process_doc   s   
z'paws_process_docs.<locals>._process_docc                 S   r   r   r   )Útextr	   r	   r
   r%   $   r   z1paws_process_docs.<locals>.lowercase_first_letterc                 S   s   | d dvo| d dvS )Nr   r   r   r	   r"   r	   r	   r
   Ú<lambda>(   s    z#paws_process_docs.<locals>.<lambda>)ÚfilterÚmap)Údatasetr&   r	   r#   r
   Úpaws_process_docs   s   ÿýr,   N)r   r   r   r,   r	   r	   r	   r
   Ú<module>   s   

