def augment_dataset()

in supervised_reptile/omniglot.py [0:0]

• 4 lines of code
• 3 McCabe index (conditional complexity)

def augment_dataset(dataset):
    """
    Augment the dataset by adding 90 degree rotations.

    Args:
      dataset: an iterable of Characters.

    Returns:
      An iterable of augmented Characters.
    """
    for character in dataset:
        for rotation in [0, 90, 180, 270]:
            yield Character(character.dir_path, rotation=rotation)