use sortkey in finish_task in distributed

mrocklin · Jul 8, 2015 · d7e0ab0 · d7e0ab0
1 parent cdbdaf0
commit d7e0ab0
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/dask/distributed/scheduler.py b/dask/distributed/scheduler.py
@@ -21,7 +21,8 @@
 
 from ..core import get_dependencies, flatten
 from .. import core
-from ..async import finish_task, start_state_from_dask as dag_state_from_dask
+from ..async import (sortkey, finish_task,
+        start_state_from_dask as dag_state_from_dask)
 
 with open('log.scheduler', 'w') as f:  # delete file
     pass
@@ -634,7 +635,7 @@ def fire_task():
                     raise payload['status']
 
                 key = payload['key']
-                finish_task(dsk, key, dag_state, results,
+                finish_task(dsk, key, dag_state, results, sortkey,
                             release_data=self._release_data)
 
                 while dag_state['ready'] and self.available_workers.qsize() > 0: