def __init__(self, *args, nb_permutations=4, **kwargs):
MNIST.__init__(self, *args, **kwargs)
self._transformations = list(range(nb_permutations))
self._mapping = None
def __init__(self, *args, angles=[45, 90, 135, 180], **kwargs):
MNIST.__init__(self, *args, **kwargs) # pylint: disable=non-parent-init-called
self._transformations = angles
self._mapping = None