o
    6ti                  	   @   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g dZ	dd Z
edkre
 Zejejd Zeejd	d
ZeeZW d   n1 sLw   Y  ee	D ]VZeejdkrfdej de nd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 sw   Y  qUdS dS )z<
Take in a YAML, and output all other splits with this YAML
    N)tqdm)WRGRRCSRCSSRCHLIc                  C   s:   t  } | jddd | jddd | jddd |  S )	Nz--base_yaml_pathT)requiredz--save_prefix_pathcsatqa)defaultz--task_prefix )argparseArgumentParseradd_argument
parse_args)parser r   Z/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/csatqa/_generate_configs.pyr      s
   r   __main__zutf-8)encodingr   csatqa__)includetaskdataset_namez.yamlzSaving yaml for subset z to winfT")widthallow_unicodedefault_style)__doc__r   loggingosyamlr   	getLogger__name__eval_loggerSUBSETSr   argspathsplitbase_yaml_pathbase_yaml_nameopenf	full_load	base_yamlnametask_prefixlower	yaml_dictsave_prefix_pathfile_save_pathinfo	yaml_filedumpfloatr   r   r   r   <module>   sH    

