Function _get_quantile_chunks

dask/array/reductions.py:1842–1847

(a, q, axis, keepdims)

def _get_quantile_chunks(a, q, axis, keepdims):
    if keepdims:
        rest = (1 if ax in axis else c for ax, c in enumerate(a.chunks))
    else:
        rest = (c for ax, c in enumerate(a.chunks) if ax not in axis)
    return [*q.shape, *rest]