Asynchronous execution of python function with ordered result

parallel.py

from multiprocessing import Pool
from time import sleep


def _wait(i):
    sleep(i)
    return i


def run_async(function, args_list, njobs, sleep_time_s = 0.01):
    pool = Pool(njobs)
    results = [pool.apply_async(function, args=args) for args in args_list]
    while not all(future.ready() for future in results):
        sleep(sleep_time_s)
    results = [result.get() for result in results]
    pool.close()
    return results
  
  
if __name__ == '__main__':
    njobs = 3
    delay = [[i] for i in range(3)]
    result = run_async(_wait, delay, njobs)
    print(result)