data/dataset_handler.py

DATASETS_KEYWORDS = [
    "train-size",
    "test-size",
    "c",
    "csharp",
    "python",
    "java",
    "javascript",
]

def get_datasets_nickname(dataset_name) -> str:
    if "csharp" == dataset_name:
        return "C#"
    elif "python" == dataset_name:
        return "Python"
    elif "javascript" == dataset_name:
        return "JavaScript"
    elif "java" == dataset_name:
        return "Java"
    elif "train-size" == dataset_name:
        return "Train\nSize"
    elif "test-size" == dataset_name:
        return "Test\nSize"
    elif "c" == dataset_name:
        return "C"
    else:
        raise ValueError(f"Dataset {dataset_name} not found")