o
    5t¾iÄ
  ã                   @   sä  d Z ddlZddlZddlZddlZddlmZ e e¡Zdddddd	d
dddddddddœZ	dd„ Z
edkrîe
ƒ Zej ej¡d ZeejddZe e¡ZW d  ƒ n1 sZw   Y  ejdur„ddlZeejddZe e¡ZW d  ƒ n1 sw   Y  ee	 ¡ ƒD ]e\ZZejdur˜ee Znde› dZeejdkr­dej› de› nde› eedœZejde› d  Ze  d!e› d"e› ¡ eed#ddZ!ej"ee!e#d$ƒd%d&d' W d  ƒ n1 sèw   Y  qŠdS dS )(z<
Take in a YAML, and output all other splits with this YAML
é    N)ÚtqdmÚpolysemy_resolutionÚpoetry_sentiment_analysisÚnamed_entity_recognitionÚbasic_ancient_chineseÚpoetry_context_predictionÚsentence_segmentationÚcouplet_predictionÚpoetry_appreciateÚancient_chinese_cultureÚancient_phoneticsÚ homographic_character_resolutionÚancient_literatureÚancient_medicalÚpoetry_quality_assessmentÚreading_comprehension)u   å¤æ–‡å•å­—å¤šä¹‰u   è¯—è¯æƒ…æ„Ÿåˆ†ç±»u   å¤æ±‰è¯­å‘½åä½“è¯†åˆ«u   å¤æ±‰è¯­çŸ¥è¯†u   å¤è¯—è¯ä¸Šä¸‹å¥é¢„æµ‹u   å¤æ–‡æ–­å¥u   å¯¹è”u   å¤è¯—è¯æ›²é‰´èµu   å›½å­¦å¸¸è¯†u	   å¤éŸ³å­¦u	   é€šå‡å­—u   å¤ä»£æ–‡å­¦çŸ¥è¯†u	   åŒ»å¤æ–‡u   å¤è¯—è¯è´¨é‡è¯„ä¼°u   å¤æ–‡é˜…è¯»ç†è§£c                  C   sH   t  ¡ } | jddd | jddd | jdd d | jdd	d |  ¡ S )
Nz--base_yaml_pathT)Úrequiredz--save_prefix_pathÚaclue)Údefaultz--cot_prompt_pathz--task_prefixÚ )ÚargparseÚArgumentParserÚadd_argumentÚ
parse_args)Úparser© r   úY/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/aclue/_generate_configs.pyr   #   s   r   Ú__main__éÿÿÿÿzutf-8)Úencodingu   ä»¥ä¸‹æ˜¯å…³äºŽu>   çš„å•é¡¹é€‰æ‹©é¢˜ï¼Œè¯·ç›´æŽ¥ç»™å‡ºæ­£ç¡®ç­”æ¡ˆçš„é€‰é¡¹ã€‚

r   Úaclue_Ú_)ÚincludeÚtaskÚdataset_nameÚdescriptionz.yamlzSaving yaml for subset z to ÚwÚinfTú")ÚwidthÚallow_unicodeÚdefault_style)$Ú__doc__r   ÚloggingÚosÚyamlr   Ú	getLoggerÚ__name__Úeval_loggerÚSUBJECTSr   ÚargsÚpathÚsplitÚbase_yaml_pathÚbase_yaml_nameÚopenÚfÚ	full_loadÚ	base_yamlÚcot_prompt_pathÚjsonÚloadÚcot_fileÚitemsÚ
subject_zhÚsubject_engr%   Útask_prefixÚ	yaml_dictÚsave_prefix_pathÚfile_save_pathÚinfoÚ	yaml_fileÚdumpÚfloatr   r   r   r   Ú<module>   sz    
ñ	ÿ
ÿ


ÿ
ÿú	ûÿ€ß