Skip to content

Commit

Permalink
[CELEBORN-1783][FOLLOWUP] Compatible with UT
Browse files Browse the repository at this point in the history
### 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 <shuaizhentao.szt@alibaba-inc.com>
Signed-off-by: Shuang <lvshuang.xjs@alibaba-inc.com>
  • Loading branch information
zaynt4606 authored and RexXiong committed Dec 19, 2024
1 parent 4b6b190 commit 6dd370d
Showing 1 changed file with 2 additions and 5 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -172,18 +172,15 @@ 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,
new util.HashSet(res.workerResource.keySet()),
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)
}
Expand Down

0 comments on commit 6dd370d

Please # to comment.