o
    6ti                     @   s    d dl mZ dd Zdd ZdS )    )weighted_f1_scorec                 C   s   d}|j | d d}|S )Na  Please provide the POS tags for each word in the input sentence. The input will be a list of words in
    the sentence. The output format should be a list of tuples, where each tuple consists of a word from the input text
    and its corresponding POS tag label from the tag label set: ["ADJ", "ADP", "ADV", "AUX", "CCONJ", "DET", "INTJ",
    "NOUN", "NUM", "PART", "PRON", "PROPN", "PUNCT" "SCONJ", "SYM", "VERB", "X"]. 
Your response should include only a
    list of tuples, in the order that the words appear in the input sentence, with each tuple containing the
    corresponding POS tag label for a word.

    Input: {tokens}
    Output: tokens)subject)format)docoutputtext r	   \/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/afrobench/masakhapos/utils.pydoc_to_text   s   
r   c                    s   i ddddddddd	d
dddddddddddddddddddddd d!d"d#d$i  fd%d&| d' D S )(Nr   NOUN   PUNCT   ADP   NUM   SYM   SCONJ   ADJ   PART   DET	   CCONJ
   PROPN   PRON   X   _   ADV   INTJ   VERB   AUXc                    s   g | ]} | qS r	   r	   ).0tagpos_tag_mapr	   r
   
<listcomp>(   s    z!doc_to_target.<locals>.<listcomp>uposr	   )r   r	   r1   r
   doc_to_target   sL   	
r5   N)lm_eval.utilsr   r   r5   r	   r	   r	   r
   <module>   s    