o
    7ti                     @   s   d dl Z d dlZdddddddd	d
ddddddddddddddddddddddddddddd d!dd"d#d$d%dd&d'd(d)dd*d+d,d-dd.d/d0d1dd2d3d4d5dd6d7d8d9dd:d;d<d=dd>Zd?ed@edAdfdBdCZdGdDdEZedFkr~e  dS dS )H    Nu   صحيحu   نعمu   لذاu   رقم)QUESTION_WORDENTAILMENT_LABELNEUTRAL_LABELCONTRADICTION_LABELu   правилноu   даu   такаu   неrichtigJaAuchNeinu   σωστόςu   Ναίu   Έτσιu   όχιrightYesAlsoNocorrectou   SízAsi quecorrectOuiAussiNonu	   सहीu	   हाँu   इसलिएu   नहींu   правильноu   Даu   Такu   НетsahihiNdiyoHivyoHapanau   ถูกต้องu	   ใช่u   ดังนั้นu	   ไม่u   doğruEvetu   Böyleceu   Hayıru   صحیحu   جی ہاںu   اس لئےu   نہیںu   đúngu   Vângu	   Vì vậyu   Khôngu   正确u   是的u   所以u	   不是的)arbgdeelenesfrhiruswthtrurvizh
output_dir	overwritereturnc           
      C   s$  g }t  D ]y}d| d}zct | d }t | d }t | d }t | d }t|  d| |r2dnd	d
d1}	|	d tjd|d| dd| d| d| d| d| d| dd|	dd W d   n1 skw   Y  W q ty   || Y qw t|dkrtdd	| dS )z
    Generate a yaml file for each language.

    :param output_dir: The directory to output the files to.
    :param overwrite: Whether to overwrite files if they already exist.
    xnli_z.yamlr   r   r   r   /wxutf8)encodingz# Generated by utils.py
xnli_common_yaml z{{[premise+", z? z, "+hypothesis,premise+", z, "+hypothesis]}})includedataset_nametaskdoc_to_textdoc_to_choiceT)allow_unicodeNr   zJFiles were not created because they already exist (use --overwrite flag): z, )
	LANGUAGESkeysopenwriteyamldumpFileExistsErrorappendlenjoin)
r'   r(   errlang	file_namer   r   r   r   f rF   L/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/xnli/utils.pygen_lang_yamlsh   s^   
rH   c                  C   sF   t  } | jddddd | jdddd	 |  }t|j|jd
 dS )z9Parse CLI args and generate language-specific yaml files.z--overwriteF
store_truez%Overwrite files if they already exist)defaultactionhelpz--output-dir.z Directory to write yaml files to)rJ   rL   )r'   r(   N)argparseArgumentParseradd_argument
parse_argsrH   r'   r(   )parserargsrF   rF   rG   main   s   rT   __main__)r)   N)rN   r<   r8   strboolrH   rT   __name__rF   rF   rF   rG   <module>   s    	^
,
