bug fix

apache · Sep 20, 2024 · 84a795a · 84a795a
1 parent 6bf19e6
commit 84a795a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/client/src/main/scala/org/apache/celeborn/client/LifecycleManager.scala b/client/src/main/scala/org/apache/celeborn/client/LifecycleManager.scala
@@ -1608,7 +1608,7 @@ class LifecycleManager(val appUniqueId: String, val conf: CelebornConf) extends
       val unregisterShuffleResponse = batchRequestMasterUnregisterShuffles(
         BatchUnregisterShuffles(appUniqueId, batchRemoveShuffleIds, MasterClient.genRequestId()))
       if (StatusCode.SUCCESS == Utils.toStatusCode(unregisterShuffleResponse.getStatus)) {
-        batchRemoveShuffleIds.forEach { shuffleId =>
+        batchRemoveShuffleIds.forEach { shuffleId: Integer =>
           unregisterShuffleTime.remove(shuffleId)
         }
       }

diff --git a/.../test/scala/org/apache/celeborn/tests/client/LifecycleManagerUnregisterShuffleSuite.scala b/.../test/scala/org/apache/celeborn/tests/client/LifecycleManagerUnregisterShuffleSuite.scala
@@ -48,7 +48,7 @@ class LifecycleManagerUnregisterShuffleSuite extends WithShuffleClientSuite
 
   test("test unregister shuffle in batch") {
     val conf = celebornConf.clone
-    conf.set(CelebornConf.CLIENT_BATCH_REMOVE_EXPIRED_SHUFFLE.key, "true")
+    conf.set(CelebornConf.CLIENT_BATCH_REMOVE_EXPIRED_SHUFFLE_ENABLED.key, "true")
     val lifecycleManager: LifecycleManager = new LifecycleManager(APP, conf)
     val counts = 10
     val ids =