o 6ti@sZddlZddlZddlZddZdedededdfdd Zd d d Zed kr+edSdS)NcCs0dd|dd|ddd|dd }||S) NzGiven the categories technology, business, politics, sports, health, entertainment, or religion; what category does the text: '{{headline}}' belong to: z Does this z topic; '{{headline}}' belong to one of the following categories: technology, business, politics, sports, health, entertainment, or religion? category only zYou are an assistant able to classify topics in texts. Given the categories technology, religion, politics, sports, health, entertainment, or business; what is the topic of the zM statement below? Return only the category. text: {{headline}} \category: zLabel the following text as technology, religion, politics, sports, health, entertainment, or geography. Provide only the category as your response. text: {{headline}} \category: zEYou are tasked with performing topic classification on the following a text. For each input, classify the topic as technology, business, politics, sports, health, entertainment, or religion. Use the following guidelines: technology: The text discusses scientific discoveries, technological advancements, or related topics. politics: The text covers political events, policies, or related topics. sports: The text talks about sports events, athletes, or related topics. health: The text addresses health issues, medical advancements, or related topics. entertainment: The text pertains to movies, music, celebrities, or related topics. religion: The text talks about relgions, religious institutions and beliefs or related topics. business: The text covers economy, business, or related topics. If the text contains multiple topics, choose the dominant topic. For ambiguous or unclear topics, select the category that best reflects the overall content. Please provide a single classification for each input. text: {{headline}} \category: prompt_1prompt_2prompt_3prompt_4prompt_5)modelang prompt_maprr]/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/afrobench/masakhanews/utils.py prompt_funcs  r output_dir overwriter returnc Csng}iddddddddd d d d d ddddddddddddddddddd }|D]l}z\d!|d"}d!|d#|}d$}|||t|||d%} tj||} tj| d&d't|d(|d(||rsd)nd*d+d,} | d-tj | | d&d.Wd/n1swYWq8t y| |Yq8wt |d0krt d1d2|d/S)3z 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. amhAmharicengEnglishfraFrenchhauHausaiboIgbolinLingalalugLugandaormz Afaan OromoopcmzNigerian PidginrunKirundisnaShonasomSomaliswaSwahilitirTigrinyaxhoXhosayorYoruba masakhanews_z.yaml_ masakhanews)includetask dataset_name doc_to_textT)exist_ok/wxutf8)encodingz# Generated by utils.py ) allow_unicodeNrzJFiles were not created because they already exist (use --overwrite flag): z, ) keysr ospathjoinmakedirsopenwriteyamldumpFileExistsErrorappendlen) rrr err languagesr file_name task_name yaml_template yaml_details file_pathfrrr gen_lang_yamls#s            rQcCspt}|jddddd|jdddd gd }|jd d ||d d|}|jD] }t|j|j|dq*dS)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)rSrUrz--mode*zPrompt number(s))nargsrSchoicesrU)rrr N)argparseArgumentParser add_argument parse_argsr rQrr)parserPROMPT_CHOICESargsr rrr mainas0 r`__main__)rN) rYr>rDr strboolrQr`__name__rrrr s >