diff --git a/common/src/main/scala/org/apache/celeborn/common/protocol/message/ControlMessages.scala b/common/src/main/scala/org/apache/celeborn/common/protocol/message/ControlMessages.scala index 7e772c74975..94d4a22a85d 100644 --- a/common/src/main/scala/org/apache/celeborn/common/protocol/message/ControlMessages.scala +++ b/common/src/main/scala/org/apache/celeborn/common/protocol/message/ControlMessages.scala @@ -785,7 +785,7 @@ object ControlMessages extends Logging { case ReportWorkerDecommission(workers, requestId) => val payload = PbReportWorkerDecommission.newBuilder() .addAllWorkers(workers.asScala.map { workerInfo => - PbSerDeUtils.toPbWorkerInfo(workerInfo, true) + PbSerDeUtils.toPbWorkerInfo(workerInfo, true, false) }.toList.asJava) .setRequestId(requestId).build().toByteArray new TransportMessage(MessageType.REPORT_WORKER_DECOMMISSION, payload) diff --git a/common/src/main/scala/org/apache/celeborn/common/util/PbSerDeUtils.scala b/common/src/main/scala/org/apache/celeborn/common/util/PbSerDeUtils.scala index e8fe6855d70..77db460ea85 100644 --- a/common/src/main/scala/org/apache/celeborn/common/util/PbSerDeUtils.scala +++ b/common/src/main/scala/org/apache/celeborn/common/util/PbSerDeUtils.scala @@ -450,7 +450,8 @@ object PbSerDeUtils { case (worker, workerEventInfo) => (worker.toUniqueId(), PbSerDeUtils.toPbWorkerEventInfo(workerEventInfo)) }.asJava) - .addAllDecommissionWorkers(decommissionWorkers.asScala.map(toPbWorkerInfo(_, true)).asJava) + .addAllDecommissionWorkers(decommissionWorkers.asScala.map( + toPbWorkerInfo(_, true, false)).asJava) if (currentAppDiskUsageMetricsSnapshot != null) { builder.setCurrentAppDiskUsageMetricsSnapshot(