"""
Mapping for dataset names to ints.
Because i had issues passing strings into seqio + t5x.
"""

T0_DS_MAPPING = {
    "super_glue_record": 0,
    "super_glue_multirc": 1,
    "super_glue_copa": 2,
    "super_glue_axg": 3,
    "super_glue_boolq": 4,
    "super_glue_rte": 5,
    "super_glue_wic": 6,
    "super_glue_wsc.fixed": 7,
    "super_glue_cb": 8,
    "rotten_tomatoes": 9,
    "glue_qqp": 10,
    "glue_mrpc": 11,
    "cosmos_qa": 12,
    "sciq": 13,
    "trec": 14,
    "wiki_qa": 15,
    "crows_pairs": 16,
    "kilt_tasks_hotpotqa": 17,
    "samsum": 18,
    "common_gen": 19,
    "paws_labeled_final": 20,
    "dbpedia_14": 21,
    "piqa": 22,
    "ropes": 23,
    "cos_e_v1.11": 24,
    "amazon_polarity": 25,
    "trivia_qa_unfiltered": 26,
    "cnn_dailymail_3.0.0": 27,
    "ai2_arc_ARC-Challenge": 28,
    "ai2_arc_ARC-Easy": 29,
    "quarel": 30,
    "gigaword": 31,
    "xsum": 32,
    "yelp_review_full": 33,
    "winogrande_winogrande_xl": 34,
    "wiki_bio": 35,
    "openbookqa_main": 36,
    "social_i_qa": 37,
    "web_questions": 38,
    "ag_news": 39,
    "qasc": 40,
    "wiki_hop_original": 41,
    "quoref": 42,
    "squad_v2": 43,
    "wiqa": 44,
    "imdb": 45,
    "race_middle": 46,
    "race_high": 47,
    "app_reviews": 48,
    "quartz": 49,
    "hellaswag": 50,
    "duorc_SelfRC": 51,
    "duorc_ParaphraseRC": 52,
    "multi_news": 53,
    "dream": 54,
    "wino_bias_type2_pro": 55,
    "wino_bias_type2_anti": 56,
    "wino_bias_type1_pro": 57,
    "wino_bias_type1_anti": 58,
    "quail": 59,
    "story_cloze_2016": 60,
    "adversarial_qa_dbert": 61,
    "adversarial_qa_dbidaf": 62,
    "adversarial_qa_droberta": 63,
    "anli": 64,
}
