Currently, if there are hru and gru dimensioned variables in the output you can't simply xr.concat them because it is only possible to specify a single dimension to concatenate along. We can improve this by handling the gru and hru dimensioned variables separately, then merging that all together after.