parallel computation with timeout handling

joblib_time.py

from time import sleep

from joblib import Parallel, delayed
from timeout_decorator import timeout, TimeoutError


@timeout(10, use_signals=False)
def f(n):
    if n == 0:
        sleep(42)
    return n


def wrapped_f(n):
    try:
        return f(n)
    except TimeoutError:
        return None


if __name__ == '__main__':
    ret = Parallel(n_jobs=-1, backend='threading')([delayed(wrapped_f)(n) for n in range(10)])
    print(ret)  # -> [None, 1, 2, 3, 4, 5, 6, 7, 8, 9]