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()) {