DEFINITIONS = {
    "medium": {
        "class_name": "CodeXGlueCcCodeRefinement",
        "dataset_type": "Code-Code",
        "description": "CodeXGLUE code-refinement dataset, available at https://github.com/microsoft/CodeXGLUE/tree/main/Code-Code/code-refinement",
        "dir_name": "code-refinement",
        "name": "medium",
        "parameters": {"size": "medium"},
        "project_url": "https://github.com/madlag/CodeXGLUE/tree/main/Code-Code/code-refinement",
        "raw_url": "https://raw.githubusercontent.com/madlag/CodeXGLUE/main/Code-Code/code-refinement/data",
        "sizes": {"test": 6545, "train": 52364, "validation": 6546},
    },
    "small": {
        "class_name": "CodeXGlueCcCodeRefinement",
        "dataset_type": "Code-Code",
        "description": "CodeXGLUE code-refinement dataset, available at https://github.com/microsoft/CodeXGLUE/tree/main/Code-Code/code-refinement",
        "dir_name": "code-refinement",
        "name": "small",
        "parameters": {"size": "small"},
        "project_url": "https://github.com/madlag/CodeXGLUE/tree/main/Code-Code/code-refinement",
        "raw_url": "https://raw.githubusercontent.com/madlag/CodeXGLUE/main/Code-Code/code-refinement/data",
        "sizes": {"test": 5835, "train": 46680, "validation": 5835},
    },
}
