o 6ti@shddlZddlZddlZGdddZddZdededed dfd d Zdd d Ze dkr2edSdS)Nc@seZdZddZdS) FunctionTagcCs ||_dS)N)value)selfrrT/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/afrixnli/gen_utils.py__init__s zFunctionTag.__init__N)__name__ __module__ __qualname__rrrrrrs rcCs,dd|dd|d|ddd}||S) NaPlease identify whether the premise entails or contradicts the hypothesis in the following premise and hypothesis. The answer should be exact entailment, contradiction, or neutral. Premise: {premise} Hypothesis: {hypothesis} Is it entailment, contradiction, or neutral?z.Given the following premise and hypothesis in z, identify if the premise entails, contradicts, or is neutral towards the hypothesis. Please respond with exact 'entailment', 'contradiction', or 'neutral'. Premise: {{premise}} Hypothesis: {{hypothesis}}zJYou are an expert in Natural Language Inference (NLI) specializing in the z7 language. Analyze the premise and hypothesis given in z, and determine the relationship between them. Respond with one of the following options: 'entailment', 'contradiction', or 'neutral'. Premise: {{premise}} Hypothesis: {{hypothesis}}zBased on the given statement, is the following claim 'true', 'false', or 'inconclusive'. Statement: {{premise}} Claim: {{hypothesis}})prompt_1prompt_3prompt_4prompt_5r)modelang prompt_maprrr prompt_func s r output_dir overwriterreturnc Csg}iddddddddd d d d d ddddddddddddddddddd d!d"d#d$i}|D]}zd%|d&}d%|d'|}d(}|d)d*d+krkd,|d&}d,|d'|}d-}t|d'd*d.kst|d'd*d/kr|||t|||d0} n|||d1} tj|d)|d2d3t|d)|d)||rd4nd5d6d7} | d8tj | | d2d9Wd:n1swYWq?t y| |Yq?wt |d;krt dz 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. engEnglishamhAmhariciboIgbofraFrenchsnachiShonawolWolofeweEwelinLingalalugLugandaxhoisiXhosakin KinyarwandatwiTwizulZuluormOromoyorYorubahauHausasotSesothoswaSwahili afrixnli_z.yaml_ afrixnli_yaml/ translateafrixnli_translate_afrixnli_translate_yaml)includetask dataset_name doc_to_text)rDrErFT)exist_okwxutf8)encodingz# Generated by utils.py ) allow_unicodeNrzJFiles were not created because they already exist (use --overwrite flag): z, )keyssplitintrosmakedirsopenwriteyamldumpFileExistsErrorappendlenjoin) rrrerr languagesr file_name task_name yaml_template yaml_detailsfrrrgen_lang_yamlss        ,     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 ./translatez Directory to write yaml files to)rdrfz--moder)r prompt_2r r rz Prompt number)rdchoicesrf)rrrN)argparseArgumentParser add_argument parse_argsrbrrr)parserargsrrrmaings(ro__main__)rN) rirQrUrrstrboolrbrorrrrrs I