Ensure store_dsk is a ShareDict at the end

Whether persisting the results to disk or building off the graph from the stored step, simply ensure that `store_dsk` is a `ShareDict`.
dask · Dec 13, 2017 · 77d586b · 77d586b
1 parent 6c6b385
commit 77d586b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/dask/array/core.py b/dask/array/core.py
@@ -924,12 +924,13 @@ def store(sources, targets, lock=True, regions=None, compute=True, keep=False, *
                 store_dlyd = persist(*store_dlyd)
                 for each_store_dlyd in store_dlyd:
                     store_dsk.update(each_store_dlyd.dask)
+                sharedict.merge(store_dsk)
             else:
-                store_dsk = dict(sharedict.merge(
+                store_dsk = sharedict.merge(
                     (store_name, store_dsk),
                     dsk,
                     src.dask
-                ))
+                )
 
             results.append(Array(
                 sharedict.merge(