python iter batch v1

python-iter-batch-v1.py

from typing import Iterator, List, TypeVar

T = TypeVar('T')


def iter_batch(sequence: Iterator[T], size: int) -> Iterator[List[T]]:
    count, buffer = 0, [None] * size
    for item in sequence:
        idx = count % size
        buffer[idx] = item
        if idx + 1 == size:
            yield buffer
        count += 1
    remaining = count % size
    if remaining > 0:
        yield buffer[: remaining]