17 lines
365 B
Python
17 lines
365 B
Python
from typing import Iterable, List, TypeVar
|
|
|
|
T = TypeVar("T")
|
|
|
|
|
|
def chunk(values: Iterable[T], chunk_size: int) -> Iterable[List[T]]:
|
|
assert chunk_size >= 1
|
|
|
|
result: List[T] = []
|
|
for v in values:
|
|
result.append(v)
|
|
if len(result) == chunk_size:
|
|
yield result
|
|
result = []
|
|
|
|
if len(result) > 0:
|
|
yield result
|