src/data/load_data.py

from datasets import load_dataset

def load_resume_dataset(train, val, test):
    return load_dataset(
        "json",
        data_files={
            "train": train,
            "validation": val,
            "test": test
        }
    )