def METHOD_NAME(path):
    # - Get train/test datasets
    if idist.get_rank() > 0:
        # Ensure that only rank 0 download the dataset
        idist.barrier()
    train_ds = datasets.CIFAR10(root=path, train=True, download=True, transform=train_transform)
    test_ds = datasets.CIFAR10(root=path, train=False, download=False, transform=test_transform)
    if idist.get_rank() == 0:
        # Ensure that only rank 0 download the dataset
        idist.barrier()
    return train_ds, test_ds