def set_axis_labels(inargs, xunits, yunits):
    """Set the x and y axis labels."""

    if inargs.xlabel:
        xname = inargs.xlabel.replace('_', ' ')
    else:
        xname = inargs.xvar.replace('_', ' ')
    xlabel = '%s (%s/yr)' %(xname, xunits)


    if inargs.ylabel:
        yname = inargs.ylabel.replace('_', ' ')
    else:
        yname = inargs.yvar[0].replace('_', ' ')

    if str(yunits) == 'kg m-2 s-1':
        yunits = '$kg \: m^{-2} \: s^{-1}$'
    ylabel = '%s (%s/yr)' %(yname, yunits)

    return xlabel, ylabel