great-ai/great_ai/utilities/unchunk.py
2022-07-04 19:31:15 +02:00

9 lines
220 B
Python

from typing import Iterable, Optional, TypeVar
T = TypeVar("T")
def unchunk(chunks: Iterable[Optional[Iterable[T]]]) -> Iterable[T]:
for chunk in chunks:
if chunk is not None:
yield from chunk