support for dask 0.18+

aertslab · Jul 12, 2018 · 264b0d3 · 264b0d3
1 parent 7c4d07b
commit 264b0d3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -7,7 +7,7 @@ frozendict
 numpy
 pandas>=0.20.1
 cloudpickle
-dask>=0.17.2
+dask>=0.18.1
 distributed>=1.21.6
 pyarrow==0.8.0
 feather-format

diff --git a/src/pyscenic/prune.py b/src/pyscenic/prune.py
@@ -19,7 +19,6 @@
 
 from boltons.iterutils import chunked_iter
 
-from dask.multiprocessing import get
 from dask import delayed
 from dask.dataframe import from_delayed
 
@@ -298,7 +297,7 @@ def wrap(data):
         # Compute dask graph ...
         if client_or_address == "dask_multiprocessing":
             # ... via multiprocessing.
-            return create_graph().compute(get=get, num_workers=num_workers if num_workers else cpu_count())
+            return create_graph().compute(scheduler='processes', num_workers=num_workers if num_workers else cpu_count())
         else:
             # ... via dask.distributed framework.
             client, shutdown_callback = _prepare_client(client_or_address, num_workers=num_workers if num_workers else cpu_count())