o 6ti @slddlZddlZddlZGdddZddZdededed ed df d d Zdd dZe dkr4edSdS)Nc@seZdZddZdS) FunctionTagcCs ||_dS)N)value)selfrr\/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/afrobench/mafand/gen_utils.py__init__s zFunctionTag.__init__N)__name__ __module__ __qualname__rrrrrrs rc Cs|d}d||d||d|dd||d||d||d |d d ||d d ||d||d |dd||d||d d}||S)N_textz]You are an advanced Translator, a specialized assistant designed to translate documents from zh into English. Your main goal is to ensure translations are grammatically correct and human-oriented. z: {{z }} English: zjYou are an advanced Translator, a specialized assistant designed to translate documents from English into zp. Your main goal is to ensure translations are grammatically correct and human-oriented. English: {eng_text} z: z sentence: {{z}} English sentence: z English sentence: {{eng_text}} z sentence: z6You are a translation expert. Translate the following z sentences to English z}} English sentence: zKYou are a translation expert. Translate the following English sentences to z" English sentence: {{eng_text}} )prompt_1prompt_1_reverseprompt_2prompt_2_reverseprompt_3prompt_3_reverser)modelang lang_dictlanguage_column_name prompt_maprrr prompt_func s:  r output_dir overwriterreversereturnc Csg}iddddddddd d d d d ddddddddddddddddddd d!d"d#d$d%d&d'}gd(}|D]}z||vrR|d)n|d*}||vr`d+|nd,|} |ri|n| } d-| d.} d-| d/|} d0} | | | d1}|r|d2|d3n|d2|d4}tj|d5d6t|d2| |rd7nd8d9d:}|d;tj||d5d<Wd=n1swYWqFty|| YqFwt |d>krtd?d@ |d=S)Az 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. amhAmharicbamBambarabbjGbomalaeweEwefonFonhauHausaiboIgbokin KinyarwandalugLugandaluoLuomosMossinyaChichewapcmzNigerian PidginsnaShonaswaSwahilitsnSetswanatwiTwiWolofXhosaYorubaZulu)wolxhoyorzul)rr r"r$rAr0z-enz-frzen-zfr-mafand_z.yaml_mafand)includetask dataset_name/z/african-englishz/english-africanT)exist_okwxutf8)encodingz# Generated by utils.py ) allow_unicodeNrzJFiles were not created because they already exist (use --overwrite flag): z, ) keysosmakedirsopenwriteyamldumpFileExistsErrorappendlenjoin)rrrrerr languages french_langsr norm_lang reverse_langrJ file_name task_name yaml_template yaml_detailsfile_dirfrrrgen_lang_yamls#s             rhcCszt}|jddddd|jdddd |jd d gd d d|jddddgdd|}t|j|j|j|jddS)z9Parse CLI args and generate language-specific yaml files.z --overwriteT store_truez%Overwrite files if they already exist)defaultactionhelpz --output-dirz./z Directory to write yaml files to)rjrlz--moder)r rrz Prompt number)rjchoicesrlz --reverseFz!Reverse the translation direction)rrrrN) argparseArgumentParser add_argument parse_argsrhrrrr)parserargsrrrmainns> rt__main__)rN) rnrSrWrrstrboolrhrtrrrrrs K$