o 6ti @slddlZddlZddlZGdddZddZdededed ed df d d Zdd dZe dkr4edSdS)Nc@seZdZddZdS) FunctionTagcCs ||_dS)N)value)selfrrZ/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/afrobench/salt/gen_utils.py__init__s zFunctionTag.__init__N)__name__ __module__ __qualname__rrrrrrs rc Cs|d}||d|dd||dd||d||d|dd ||d ||dd ||d ||d ||d|d d ||d||d||dd}||S)N_textz sentence: {{z}} English sentence: z'English sentence: {{eng_source_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_source_text}} zAs a z/ and English linguist, translate the following z sentences to English. zD and English linguist, translate the following English sentences to z*. English sentence: {{eng_source_text}} )prompt_1prompt_1_reverseprompt_2prompt_2_reverseprompt_3prompt_3_reverser)modelang lang_dictlanguage_column_name prompt_maprrr prompt_func s6   r output_dir overwriterreversereturnc Csg}ddddddddd }|D]}z|d kr|ssd |d }d |d |}d} | |ddt|||d} tj|d|ddt|d|d||rPdnddd} | dtj| | ddWdn1smwYn^d|d}d|d|}d} | |d|dt|d ||d} tj|d|ddt|d|d||rdnddd} | dtj| | ddWdn1swYWqty| |Yqwt |d!krtd"d# |dS)$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. EnglishLugandaAcholiLugbaraAteso RunyankoleSwahiliIgbo)englugachlggteonynswaibor$salt_z -eng.yamlz-eng_saltztext-alleng_target_text)includetask dataset_name doc_to_target doc_to_text/T)exist_okwxutf8)encodingz# Generated by utils.py ) allow_unicodeNz salt_eng-z.yaml_r _reverserzJFiles were not created because they already exist (use --overwrite flag): z, ) keysrosmakedirsopenwriteyamldumpFileExistsErrorappendlenjoin) rrrrerr languagesr file_name task_name yaml_template yaml_detailsfrrrgen_lang_yamls"s          rOcCszt}|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)rQrSz--moder )r rrz Prompt number)rQchoicesrSz --reverseFz!Reverse the translation direction)rrrrN) argparseArgumentParser add_argument parse_argsrOrrrr)parserargsrrrmainps> r[__main__)rN) rUr>rBrrstrboolrOr[rrrrrs N$