Since the forecast methods are just passed an numpy array and they return another numpy array, it might be possible to just wrap them in a method that extracts the numpy array from the xarray dataset and then puts the result in a new xarray with the correct metadata.