class RandomFlip(grain.RandomMapTransform):
    def random_map(self, element, rng):
        if rng.random() > 0.5:
            element['image'] = np.fliplr(element['image'])
        return element

        __
        __