From cefbb0c94c0be793fd1f5eac44e0f8b844fa48ec Mon Sep 17 00:00:00 2001
From: Hao Chen <chenh1024@gmail.com>
Date: Fri, 28 Jun 2019 12:56:48 +0800
Subject: [PATCH] Fix driver id in TaskInfo (#5055)

---
 src/ray/core_worker/common.h          | 4 ++--
 src/ray/core_worker/task_execution.cc | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/ray/core_worker/common.h b/src/ray/core_worker/common.h
index b5288c033..9dbdfdc77 100644
--- a/src/ray/core_worker/common.h
+++ b/src/ray/core_worker/common.h
@@ -74,8 +74,8 @@ enum class TaskType { NORMAL_TASK, ACTOR_CREATION_TASK, ACTOR_TASK };
 struct TaskInfo {
   /// The ID of task.
   const TaskID task_id;
-  /// The driver ID.
-  const DriverID driver_id;
+  /// The job ID.
+  const JobID job_id;
   /// The type of task.
   const TaskType task_type;
 };
diff --git a/src/ray/core_worker/task_execution.cc b/src/ray/core_worker/task_execution.cc
index 68ab69299..fd402fa33 100644
--- a/src/ray/core_worker/task_execution.cc
+++ b/src/ray/core_worker/task_execution.cc
@@ -46,7 +46,7 @@ Status CoreWorkerTaskExecutionInterface::Run(const TaskExecutor &executor) {
         task_type = TaskType::NORMAL_TASK;
       }
 
-      TaskInfo task_info{spec.TaskId(), spec.DriverId(), task_type};
+      TaskInfo task_info{spec.TaskId(), spec.JobId(), task_type};
 
       auto num_returns = spec.NumReturns();
       if (spec.IsActorCreationTask() || spec.IsActorTask()) {