[CELEBORN-1337][FOLLOWUP] Fix compile problem

### What changes were proposed in this pull request? Fix compile problem as CELEBORN-1337 auto merge conflict with CELEBORN-1444. ### Why are the changes needed? Fix compile problem as 1337 conflict with CELEBORN-1444. ### Does this PR introduce _any_ user-facing change? NO ### How was this patch tested? PASS GA Closes #2559 from RexXiong/minor_fix_compile_problem. Authored-by: Shuang <[email protected]> Signed-off-by: SteNicholas <[email protected]>
apache · Jun 13, 2024 · 0298cfb · 0298cfb
1 parent 5323c1d
commit 0298cfb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
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
@@ -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
@@ -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(