wrong chunk sizes in html repr with nonuniform chunks

What happened:

The HTML repr is using the first element in a chunks tuple;

What you expected to happen:

it should be using whatever dask does in this case

Minimal Complete Verifiable Example:

import xarray as xr
import dask


test = xr.DataArray(
    dask.array.zeros(
        (12, 901, 1001),
        chunks=(
            (1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1),
            (1, 899, 1),
            (1, 199, 1, 199, 1, 199, 1, 199, 1, 199, 1),
        ),
    )
)
test.to_dataset(name="a")

EDIT: The text repr has the same issue

<xarray.Dataset>
Dimensions:  (dim_0: 12, dim_1: 901, dim_2: 1001)
Dimensions without coordinates: dim_0, dim_1, dim_2
Data variables:
    a        (dim_0, dim_1, dim_2) float64 dask.array<chunksize=(1, 1, 1), meta=np.ndarray>

Start by running the issue's Python example and inspect the DataArray and Dataset HTML and text representations, focusing on how nonuniform chunk metadata is displayed. Compare the reported chunk sizes with the original chunks tuple; done means both representations show the correct chunk sizes for every dimension.