batch_generator.py

def generate_batch(generator, batch_size=10):
    batch = []
    for e in generator:
        if len(batch) == batch_size:
            yield batch
            batch = []
        batch.append(e)
    if batch:
        yield batch


if __name__ == '__main__':
    for batch in generate_batch(range(23), 10):
        print(batch)