o 6ti @shddlZddlZddlZGdddZddZdededed dfd d Zdd d Ze dkr2edSdS)Nc@seZdZddZdS) FunctionTagcCs ||_dS)N)value)selfrrY/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/afrobench/adr/gen_utils.py__init__s zFunctionTag.__init__N)__name__ __module__ __qualname__rrrrrrs rc CsBddd|dd|d|dd|d|d |d d }||S) NzfPlease restore the missing diacritics in the following sentence: {{text}}. Return output sentence onlyzGiven a sentence without diacritics, add the appropriate diacritics to make it grammatically and semantically correct. Sentence: {{text}}. Return output sentence onlyzThis text is in zw. Restore all diacritical marks to their proper places in the following sentence: {{text}}. Return output sentence onlyz9You are a linguist specializing in diacritical marks for z). Add the appropriate diacritics to this z0 sentence: {{text}}. Return output sentence onlyzC. Diacritics are essential for proper pronunciation and meaning in z!. You are tasked with converting z{ sentences without diacritics into their correctly accented forms. Here's the input: {{text}}. Return output sentence onlyprompt_1prompt_2prompt_3prompt_4prompt_5r)modelang prompt_maprrr prompt_func s r output_dir overwriterreturnc Csg}dddddd}|D]j}zZd|d}d|d |}d }|||t|||d } tj|d |d dt|d |d ||rGdnddd} | dtj| | d dWdn1sdwYWqtyx| |Yqwt |dkrtdd |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. FonGbomalaIgboWolofYoruba)fonbbjibowolyorafridiacritics_z.yaml_afridiacritics_yaml)includetask dataset_name doc_to_text/T)exist_okwxutf8)encodingz# Generated by utils.py ) allow_unicodeNrzJFiles were not created because they already exist (use --overwrite flag): z, ) keysrosmakedirsopenwriteyamldumpFileExistsErrorappendlenjoin) rrrerr languagesr file_name task_name yaml_template yaml_detailsfrrrgen_lang_yamlssT       rBcCs`t}|jddddd|jdddd |jd d gd d d|}t|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)rDrFz--moder r z Prompt number)rDchoicesrF)rrrN)argparseArgumentParser add_argument parse_argsrBrrr)parserargsrrrmainOs(rN__main__)rN) rHr1r5rrstrboolrBrNrrrrrs 1