batch_generator2.py

from itertools import groupby
from operator import itemgetter


def generate_batch(generator, batch_size=10):
    if batch_size <= 1:
        yield from generator
    mod_generator = list((i // batch_size, v) for i, v in enumerate(generator))

    for mod, tpl in groupby(mod_generator, itemgetter(0)):
        yield (v for _, v in tpl)


if __name__ == '__main__':
    for batch in generate_batch('abcdef', 2):
        print(list(batch))