def ensemble_mean(ds, ensemble_dim="member"):
    """Return the ensemble mean of the input array

    Parameters
    ----------
    ds : xarray Dataset
        Array to take the ensemble mean of
    ensemble_dim : str, optional
        The name of the ensemble dimension
    """
    return ds.mean(ensemble_dim)