From 6dd370d5b02fb0bfc1361114f7833a7ab2177d52 Mon Sep 17 00:00:00 2001 From: zhengtao Date: Thu, 19 Dec 2024 17:43:23 +0800 Subject: [PATCH] [CELEBORN-1783][FOLLOWUP] Compatible with UT ### What changes were proposed in this pull request? Compatible with UT for branch-0.5 ### Why are the changes needed? ### Does this PR introduce _any_ user-facing change? ### How was this patch tested? Closes #3011 from zaynt4606/branch-0.5-dev. Authored-by: zhengtao Signed-off-by: Shuang --- .../tests/client/LifecycleManagerCommitFilesSuite.scala | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/tests/spark-it/src/test/scala/org/apache/celeborn/tests/client/LifecycleManagerCommitFilesSuite.scala b/tests/spark-it/src/test/scala/org/apache/celeborn/tests/client/LifecycleManagerCommitFilesSuite.scala index 85226f3dd1e..959947d8fd2 100644 --- a/tests/spark-it/src/test/scala/org/apache/celeborn/tests/client/LifecycleManagerCommitFilesSuite.scala +++ b/tests/spark-it/src/test/scala/org/apache/celeborn/tests/client/LifecycleManagerCommitFilesSuite.scala @@ -172,10 +172,7 @@ class LifecycleManagerCommitFilesSuite extends WithShuffleClientSuite with MiniC val res = lifecycleManager.requestMasterRequestSlotsWithRetry(shuffleId, ids) assert(res.status == StatusCode.SUCCESS) - lifecycleManager.setupEndpoints( - res.workerResource.keySet(), - shuffleId, - new ShuffleFailedWorkers()) + lifecycleManager.setupEndpoints(res.workerResource, shuffleId, new ShuffleFailedWorkers()) lifecycleManager.reserveSlotsWithRetry( shuffleId, @@ -183,7 +180,7 @@ class LifecycleManagerCommitFilesSuite extends WithShuffleClientSuite with MiniC res.workerResource, updateEpoch = false) - lifecycleManager.commitManager.registerShuffle(shuffleId, 1, false) + lifecycleManager.commitManager.registerShuffle(shuffleId, 1) 0 until 1000 foreach { partitionId => lifecycleManager.commitManager.finishMapperAttempt(shuffleId, 0, 0, 1, partitionId) }