Skip to content

Commit 82e01c2

Browse files
committed
use safeRemoveExecutor in doKillExecutors
Signed-off-by: Sudipto Baral <sudiptobaral.me@gmail.com>
1 parent 6732dc9 commit 82e01c2

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/main/scala/org/apache/spark/scheduler/cluster/armada/ArmadaClusterManagerBackend.scala

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,6 @@ import org.apache.spark.scheduler.{
4040
ExecutorDecommission,
4141
ExecutorDecommissionInfo,
4242
ExecutorExited,
43-
ExecutorKilled,
4443
TaskSchedulerImpl
4544
}
4645
import org.apache.spark.scheduler.cluster.{CoarseGrainedSchedulerBackend, SchedulerBackendUtils}
@@ -359,7 +358,10 @@ private[spark] class ArmadaClusterManagerBackend(
359358
// Send RPC kill signal to executors
360359
executorIds.foreach { id =>
361360
markTerminal(id)
362-
removeExecutor(id, ExecutorKilled)
361+
safeRemoveExecutor(
362+
id,
363+
ExecutorExited(-1, exitCausedByApp = false, "Executor killed by Spark")
364+
)
363365
}
364366

365367
// Cancel Armada jobs after grace period

0 commit comments

Comments
 (0)