This is an automated email from the ASF dual-hosted git repository.
rexxiong pushed a commit to branch branch-0.5
in repository https://gitbox.apache.org/repos/asf/celeborn.git
The following commit(s) were added to refs/heads/branch-0.5 by this push:
new 6dd370d5b [CELEBORN-1783][FOLLOWUP] Compatible with UT
6dd370d5b is described below
commit 6dd370d5b02fb0bfc1361114f7833a7ab2177d52
Author: zhengtao <[email protected]>
AuthorDate: Thu Dec 19 17:43:23 2024 +0800
[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 <[email protected]>
Signed-off-by: Shuang <[email protected]>
---
.../celeborn/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 85226f3dd..959947d8f 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)
}
