o
    6tiq                     @   s~  d dl Z d dlZG dd dZi 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	d2d3d4d5d6d	d7d8d9d:d;d	d<d=d>d?d@d	dAdBdCdDdEd	dFdGdHdIdJd	dKdLdMdNdOd	dPdQdRdSdTd	dUdVdWdXdYd	dZd[d\d]d^d	iZd_ed`edaedbdfdcddZdhdedfZedgkre  dS dS )i    Nc                   @   s   e Zd Zdd ZdS )FunctionTagc                 C   s
   || _ d S )N)value)selfr    r   P/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/afrixnli/utils.py__init__   s   
zFunctionTag.__init__N)__name__
__module____qualname__r   r   r   r   r   r      s    r   amhu   ትክክልu   አዎu   እንዲሁምu   አይ)QUESTION_WORDENTAILMENT_LABELNEUTRAL_LABELCONTRADICTION_LABELengRightYesAlsoNoeweu   Esɔ gbeu   Ɛ̃u   HãAofracorrectOuiAussiNonhauDaidaiEez	Haka kumazA'aibozZiri eziu   Éèu
   ỌzọkwaMbakinNibyoYegozNa noneOyalinMalamuIyoLisusuTelugKituufuYeeu   N’ekiralaNeddaormSirriiEeyyee	AkkasumasLakkisnaChokwadiHonguUyeKwetesotNepileEHapeTjheswaSahihiNdiyoPiaHapanatwiNifaAaneu   AnaasɛDaabiwolu   DëggWaawItamu   Déetxho
IchanekileEwe	KananjaloHayiyoru
   Òótọ́u   Bẹ́ẹ̀niu   Àti péu   Rárázul	KulungileYeboFuthiCha
output_dir	overwritemodereturnc                 C   s  g }g d}|D ]}z|dkrzt | d }t | d }t | d }t | d }	d| d}
d| }d	}t|  d
|
 |r?dnddd-}|d tj|||d| d| d| d| d| d|	 dd|dd W d   n1 stw   Y  nHd| d| d}
d| d| }d| d}t|  d
|
 |rdnddd}|d tj|||d|dd W d   n1 sw   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.
    )r   r   r   r   r3   rF   r   r%   r*   rI   r!   rB   rO   r.   rN   r   r8   r=   native-directr   r   r   r   afrixnli_native_direct_z.yamlafrixnli_native_direct_yaml/wxutf8)encodingz# Generated by utils.py
z{{[premise+", z? z, "+hypothesis,premise+", z, "+hypothesis]}})includetaskdataset_namedoc_to_choiceT)allow_unicodeN	afrixnli___yaml)r`   ra   rb   r   zJFiles were not created because they already exist (use --overwrite flag): z, )		LANGUAGESopenwriteyamldumpFileExistsErrorappendlenjoin)rT   rU   rV   err	languageslangr   r   r   r   	file_name	task_nameyaml_templatefr   r   r   gen_lang_yamls{   s   





rx   c                  C   s`   t  } | jddddd | jdddd	 | jd
dg ddd |  }t|j|j|jd dS )z9Parse CLI args and generate language-specific yaml files.z--overwriteT
store_truez%Overwrite files if they already exist)defaultactionhelpz--output-dirz./manual/translatez Directory to write yaml files to)rz   r|   z--modemanual_translate)	en_directrX   manual_directr}   zMode of chain-of-thought)rz   choicesr|   )rT   rU   rV   N)argparseArgumentParseradd_argument
parse_argsrx   rT   rU   rV   )parserargsr   r   r   main   s(   r   __main__)rW   N)	r   rk   r   rh   strboolrx   r   r   r   r   r   r   <module>   s   %+17=CIOU[agp
X
