[xray] Unsubscribe to task dependencies when task starts execution (#2354)

* Add back call to unsubscribe to task dependencies

* fix
This commit is contained in:
Stephanie Wang
2018-07-05 21:08:58 -07:00
committed by GitHub
parent c50f1966e0
commit 5b7475a2e0
+3
View File
@@ -809,6 +809,9 @@ void NodeManager::AssignTask(Task &task) {
lineage_cache_.AddReadyTask(task);
// Mark the task as running.
local_queues_.QueueRunningTasks(std::vector<Task>({task}));
// Notify the task dependency manager that we no longer need this task's
// object dependencies.
task_dependency_manager_.UnsubscribeDependencies(spec.TaskId());
} else {
RAY_LOG(WARNING) << "Failed to send task to worker, disconnecting client";
// We failed to send the task to the worker, so disconnect the worker.