Function worker

dali/test/python/test_external_source_parallel_shared_batch.py:70–77

(start_method, sock, task_queue, res_queue, worker_cb, worker_params)

def worker(start_method, sock, task_queue, res_queue, worker_cb, worker_params):
    if start_method == "spawn":
        task_queue.open_shm(multiprocessing.reduction.recv_handle(sock))
        res_queue.open_shm(multiprocessing.reduction.recv_handle(sock))
        sock.close()
    while True:
        if worker_cb(task_queue, res_queue, **worker_params) is None:
            break


@contextmanager