def no_augmentation(x, y=None, epoch=0, rng=None, **kwargs):
    """If the trainer isn't supplied a function for data augmentation,
    use this blank function instead."""
    if y is None:
        return x
    else:
        return x, y