From 5b7475a2e0d73fa1dee44884800257349a13fcd6 Mon Sep 17 00:00:00 2001
From: Stephanie Wang <swang@cs.berkeley.edu>
Date: Thu, 5 Jul 2018 21:08:58 -0700
Subject: [PATCH] [xray] Unsubscribe to task dependencies when task starts
 execution (#2354)

* Add back call to unsubscribe to task dependencies

* fix
---
 src/ray/raylet/node_manager.cc | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/ray/raylet/node_manager.cc b/src/ray/raylet/node_manager.cc
index b9eb500a5..10825edba 100644
--- a/src/ray/raylet/node_manager.cc
+++ b/src/ray/raylet/node_manager.cc
@@ -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.