num-workers kwargs passes through to threaded get

mrocklin · Aug 3, 2015 · 616b77f · 616b77f
1 parent 1ab72aa
commit 616b77f
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/dask/threaded.py b/dask/threaded.py
@@ -19,7 +19,7 @@ def _thread_get_id():
     return current_thread().ident
 
 
-def get(dsk, result, cache=None, **kwargs):
+def get(dsk, result, cache=None, num_workers=None, **kwargs):
     """ Threaded cached implementation of dask.get
 
     Parameters
@@ -46,7 +46,10 @@ def get(dsk, result, cache=None, **kwargs):
     pool = _globals['pool']
 
     if pool is None:
-        pool = default_pool
+        if num_workers:
+            pool = ThreadPool(num_workers)
+        else:
+            pool = default_pool
 
     queue = Queue()
     results = get_async(pool.apply_async, len(pool._pool), dsk, result,