Function function_with_timeout

applications/lats/humaneval/timeout_utils.py:22–31

(func, args, timeout)

def function_with_timeout(func, args, timeout):
    r = []
    def w():
        r.append(func(*args))
    t = _PropagatingThread(target=w)
    t.start()
    t.join(timeout)
    if t.is_alive():
        raise TimeoutError()
    return r[0]