heat_demand with chunked datasets about atlite HOT 2 CLOSED

import atlite
import matplotlib.pyplot as plt
import geopandas as gpd
from pathlib import Path
import xarray as xr


pypsa_eur_dir = Path('/home/fabian/vres/py/pypsa-eur/')

chunked = atlite.Cutout('pypsa-eur.nc', chunks={'time':100})
non_chunked = atlite.Cutout('pypsa-eur.nc', chunks=None)

layout = xr.ones_like(chunked.data.height)
kwargs = dict(layout=layout, hour_shift=3)
heat_demand = xr.Dataset(dict(chunked = chunked.heat_demand(**kwargs),
                              non_chunked = non_chunked.heat_demand(**kwargs)))

fig, (ax1, ax2) = plt.subplots(2,1, sharex=True)
heat_demand.to_array(name='Total Heat Demand Europe').plot.line(x='time', ax=ax1)
(chunked.data.temperature.mean(['x', 'y']).resample(time='1D').mean() - 273)\
        .rename('T [°C]').plot.line(ax=ax2)
fig.tight_layout()
fig.savefig('heat_demand_chunked.png')