def load_data(base: str = "data/Baltruschat") -> dict:
    """Helper function that takes path to working directory and
    returns a dictionary containing the paths to the training and testsets.

    Parameters
    ----------
    base
        path to folder containing dataset sd files

    Returns
    ----------
    dict
        keys (str): dataset names
        values (str): dataset abs. paths

    """
    sdf_filepath_training = f"{base}/combined_training_datasets_unique.sdf"
    sdf_filepath_novartis = f"{base}/novartis_cleaned_mono_unique_notraindata.sdf"
    sdf_filepath_Literture = f"{base}/AvLiLuMoVe_cleaned_mono_unique_notraindata.sdf"

    datasets = {
        "Training": sdf_filepath_training,
        "Novartis": sdf_filepath_novartis,
        "Literature": sdf_filepath_Literture,
    }
    return datasets