B F.\@sfddlmZmZmZddlmZddlZddlmZm Z ddl m Z m Z m Z ed d d Zd d ZdS))print_functiondivisionabsolute_import)contextmanagerN)secederejoin) thread_state get_client get_workerTccsBt}t|d}|r.t|j|jtjd|V|r>tdS)a Get client for this thread This context manager is intended to be called within functions that we run on workers. When run as a context manager it delivers a client ``Client`` object that can submit other tasks directly from that worker. Parameters ---------- timeout: Number Timeout after which to err separate_thread: bool, optional Whether to run this function outside of the normal thread pool defaults to True Examples -------- >>> def func(x): ... with worker_client() as c: # connect from worker back to scheduler ... a = c.submit(inc, x) # this task can submit more tasks ... b = c.submit(dec, x) ... result = c.gather([a, b]) # and gather results ... return result >>> future = client.submit(func, 1) # submit func(1) on cluster See Also -------- get_worker get_client secede )timeoutz long-runningN) r r rZloopZ add_callbackZ transitionr keyr)r Zseparate_threadworkerZclientr8lib/python3.7/site-packages/distributed/worker_client.py worker_client s! rcOstdt||S)Nz'local_client has moved to worker_client)warningswarnr)argskwargsrrr local_client7s r)r T)Z __future__rrr contextlibrrZthreadpoolexecutorrrrr r r rrrrrrs  ,