def prox_lim(xy, step, boundary=None):
    """Proximal projection operator"""
    if boundary == "circle":
        return prox_circle(xy, step)
    if boundary == "line":
        return prox_line(xy, step)
    # default: do nothing
    return xy
Proximal projection operator