def get_dataset(dataloaders: dict) -> dict:
    """
    From dictionary of dataloaders to dictionary of datasets
    """
    datasets = {split: dataloader.dataset for split, dataloader in dataloaders}
    return datasets