Generate iterable chunk by batch_size # noqa: DAR301
(instances: Iterable[Any], batch_size: int | None = None)
| 47 | |
| 48 | |
| 49 | def chunk(instances: Iterable[Any], batch_size: int | None = None) -> Iterable[Iterable[Any]]: |
| 50 | """ |
| 51 | Generate iterable chunk by batch_size |
| 52 | # noqa: DAR301 |
| 53 | """ |
| 54 | if not batch_size: |
| 55 | yield instances |
| 56 | else: |
| 57 | yield from batched(instances, batch_size) |
no test coverage detected
searching dependent graphs…