o
    7ti[                     @   sH   d Z ddlmZ g Zddddddd	d
ZdddZedkr"e  dS dS )zb
Script to generate task YAMLs for the mlqa dataset.
Based on `tasks/bigbench/generate_tasks.py`.
    )get_dataset_config_namesenglishspanishhindi
vietnamesegermanarabicchinese)eneshividearzhreturnNc               	   C   s   t ddd} | D ]}t|ddkrqt| qttdks"J tD ]R}|dd d}|dd	 }t|d
dd1}|d |d |d|dd d |d| d |d| d W d    n1 sqw   Y  q$d S )Nzfacebook/mlqaT)trust_remote_code.   1   _z.yaml   wzutf-8)encodingz!# Generated by generate_tasks.py
zinclude: mlqa_common_yaml
ztask: 
zdataset_name: z1process_results: !function utils.process_results_)r   lensplitchosen_subtasksappendreplaceopenwrite)configsconfigtask	file_namecontext_langf r(   U/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/mlqa/generate_tasks.pymain   s(   


r*   __main__)r   N)__doc__datasetsr   r   language_dictr*   __name__r(   r(   r(   r)   <module>   s   

