def fix_dims_for_output(forecast_dataset):
    """Manipulate the dimensions of the dataset of a single forecast so that we
    can concatenate them easily."""

    return (
        forecast_dataset.stack(
            {"forecast_label": ["forecast_year", "forecast_monthday"]}
        )
        .expand_dims("forecast_time")
        .drop("forecast_label")
        .squeeze("forecast_label")
    )