o
    6ti                     @   s   d Z ddlm  mZ ddlZddlmZ e ZdgZ	e	D ]@Z
eeD ]9Zz)ede de
 d ejdee
dZed	e de
 d ed
ed  [W q$ ey] Z zedZ[ww qdS )a  
A utility script that pushes all Bigbench subtasks from their form in the `bigbench` HF dataset
into `{org name}/bigbench`.

Prior to running, log into HF Hub for the target HF hub org via `huggingface-cli login`.

Requires the installation of
`pip install "bigbench @ https://storage.googleapis.com/public_research_data/bigbench/bigbench-0.0.1.tar.gz"`
and is included so that the bigbench dependency can be avoided.
    N)tqdmz	Loading 'z' with num_shots=z...bigbench)name	num_shotsz	Pushing 'zhails/bigbench
_zero_shot)__doc__bigbench.api.utilapiutilbb_utilsdatasetsr   get_all_json_task_namesall_task_namesr   shots	task_nameprintload_datasettask_dspush_to_hub	Exceptione r   r   `/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/bigbench/push_bigbench_dataset.py<module>   s(    