def METHOD_NAME(r=None, nr=100):
    """
    Returns the final temperature profile.
    :param r:      Minor radius (m). May be either scalar or numpy.ndarray.
    :param int nr: Radial grid resolution to use if ``r = None``.
    """
    global a
    if r is None:
        r = np.linspace(0, a, nr)
    elif type(r) == float or type(r) == int:
        r = np.array([float(r)])
    # Flat temperature profile
    return r, 50*np.ones(r.shape)