Inform workers when results are stolen back
@@ -2187,8 +2187,8 @@ class Scheduler(Server):
logger.info("Unexpected worker completed task, likely due to"
" work stealing. Expected: %s, Got: %s, Key: %s",
w, worker, key)
- # msg = {'op': 'release-task', 'key': key}
- # self.worker_comms[w].send(msg)
+ msg = {'op': 'release-task', 'key': key, 'reason': 'stolen'}
+ self.worker_comms[w].send(msg)
recommendations = OrderedDict()
