FutureWarning: The return type of `Dataset.dims` will be changed to return a set of dimension names in future, in order to be more consistent with `DataArray.dims`. To access a mapping from dimension names to lengths, please use `Dataset.sizes`.
shape_chunked_only = np.array([ds.dims[dim] for dim in dims_chunked_only])
Get rid of the