# KLUE
from typing import Dict


class ReKlueLabels:
    SIZE: int = 30
    LABEL2IDX: Dict[str, int] = {
        "no_relation": 0,
        "org:dissolved": 1,
        "org:founded": 2,
        "org:place_of_headquarters": 3,
        "org:alternate_names": 4,
        "org:member_of": 5,
        "org:members": 6,
        "org:political/religious_affiliation": 7,
        "org:product": 8,
        "org:founded_by": 9,
        "org:top_members/employees": 10,
        "org:number_of_employees/members": 11,
        "per:date_of_birth": 12,
        "per:date_of_death": 13,
        "per:place_of_birth": 14,
        "per:place_of_death": 15,
        "per:place_of_residence": 16,
        "per:origin": 17,
        "per:employee_of": 18,
        "per:schools_attended": 19,
        "per:alternate_names": 20,
        "per:parents": 21,
        "per:children": 22,
        "per:siblings": 23,
        "per:spouse": 24,
        "per:other_family": 25,
        "per:colleagues": 26,
        "per:product": 27,
        "per:religion": 28,
        "per:title": 29,
    }
    IDX2LABEL: Dict[str, str] = {
        "0": "no_relation",
        "1": "org:dissolved",
        "2": "org:founded",
        "3": "org:place_of_headquarters",
        "4": "org:alternate_names",
        "5": "org:member_of",
        "6": "org:members",
        "7": "org:political/religious_affiliation",
        "8": "org:product",
        "9": "org:founded_by",
        "10": "org:top_members/employees",
        "11": "org:number_of_employees/members",
        "12": "per:date_of_birth",
        "13": "per:date_of_death",
        "14": "per:place_of_birth",
        "15": "per:place_of_death",
        "16": "per:place_of_residence",
        "17": "per:origin",
        "18": "per:employee_of",
        "19": "per:schools_attended",
        "20": "per:alternate_names",
        "21": "per:parents",
        "22": "per:children",
        "23": "per:siblings",
        "24": "per:spouse",
        "25": "per:other_family",
        "26": "per:colleagues",
        "27": "per:product",
        "28": "per:religion",
        "29": "per:title",
    }
