add worker_client to dask.distributed api

mrocklin · May 4, 2017 · d8bf762 · d8bf762
1 parent bd5f3ef
commit d8bf762
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dask/distributed.py b/dask/distributed.py
@@ -4,7 +4,7 @@
 try:
     from distributed import (Client, Executor, progress, as_completed, wait,
             LocalCluster, Scheduler, Worker, nanny, local_client, config,
-            CompatibleExecutor)
+            CompatibleExecutor, worker_client)
 except ImportError:
     msg = ("Dask's distributed scheduler is not installed.\n\n"
            "Please either conda or pip install dask distributed:\n\n"