build_ds.py

@dask.delayed
def load_cog(url):
    return rxr.open_rasterio(url, 
                            chunks=({'x': 512, 'y': 512}), 
                            lock=False,
                            mask_and_scale=True).squeeze('band', drop=True)

@dask.delayed
def load_scene(scene_links):
    delayed_loads = [load_cog(url) for url in scene_links] 
    ds = xr.concat(dask.compute(*delayed_loads), dim="band")
    ds.coords['band'] = BANDS
    return ds

delayed_ds = [load_scene(scene_links) for scene_links in granule_links]

time_var = xr.Variable('time', time)
ts = xr.concat(dask.compute(*delayed_ds), dim=time_var)
ts