Function grid_rechunk

xarray/backends/chunks.py:156–181

(
    v: Variable,
    enc_chunks: tuple[int, ...],
    region: tuple[slice, ...],
)

def grid_rechunk(
    v: Variable,
    enc_chunks: tuple[int, ...],
    region: tuple[slice, ...],
) -> Variable:
    nd_v_chunks = v.chunks
    if not nd_v_chunks:
        return v

    nd_grid_chunks = tuple(
        build_grid_chunks(
            v_size,
            region=interval,
            chunk_size=chunk_size,
        )
        for v_size, chunk_size, interval in zip(
            v.shape, enc_chunks, region, strict=True
        )
    )

    nd_aligned_chunks = align_nd_chunks(
        nd_v_chunks=nd_v_chunks,
        nd_backend_chunks=nd_grid_chunks,
    )
    v = v.chunk(dict(zip(v.dims, nd_aligned_chunks, strict=True)))
    return v


def validate_grid_chunks_alignment(