o 6ti@sddlZddlZddlZddlZejdddZeejZddedDZ de d e d dfd d Z dd dZ edkr?e dSdS)NzEhttps://datasets-server.huggingface.co/splits?dataset=HiTZ%2FEusExams)timeoutcCsg|]}|dqS)config).0splitrrS/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/eus_exams/configs.py sr splits output_dir overwritereturnc Csg}tD]N}d|d}z8t|d||rdnd }|dtd|vr)dnd|d|d |Wd n1s>wYWqtyR||Yqwt|d krctd d |d S)z Generate a yaml file for each configuage. :param output_dir: The directory to output the files to. :param overwrite: Whether to overwrite files if they already exist. eus_exams_z.yaml/wxz# Generated by utils.py eus_exams_es eus_exams_eu)include dataset_nametaskNrzJFiles were not created because they already exist (use --overwrite flag): z, ) CONFIGSopenwriteyamldumpFileExistsErrorappendlenjoin)r r errr file_namefrrrgen_config_yamlss6    r#cCsFt}|jddddd|jdddd |}t|j|jd d S) z;Parse CLI args and generate configuage-specific yaml files.z --overwriteF store_truez%Overwrite files if they already exist)defaultactionhelpz --output-dir.z Directory to write yaml files to)r%r')r r N)argparseArgumentParser add_argument parse_argsr#r r )parserargsrrrmain1sr/__main__)r N)r)jsonrequestsrgetresponseloadstext response_jsonrstrboolr#r/__name__rrrrs  !