For now, take a checked in head and budget output file and use imod-python to read them with lazy/dask.