def _chunk_seq()

in gym3/concat.py [0:0]

• 9 lines of code
• 2 McCabe index (conditional complexity)

def _chunk_seq(x: Sequence, sizes: Sequence[int]) -> Sequence[Any]:
    result = []
    assert len(x) == sum(sizes), f"x has incorrect length {len(x)} != {sum(sizes)}"
    start = 0
    for size in sizes:
        end = start + size
        result.append(x[start:end])
        start = end
    return result