Resubmitting a stage without verifying the stage attempt number may result in an infinite loop

private def processShuffleMapStageCompletion(shuffleStage: ShuffleMapStage): Unit = {
    // some code ... 
    
    if (!shuffleStage.isAvailable) {
      // Some tasks had failed; let's resubmit this shuffleStage.
      // TODO: Lower-level scheduler should also deal with this
      logInfo(log"Resubmitting ${MDC(STAGE, shuffleStage)} " +
        log"(${MDC(STAGE_NAME, shuffleStage.name)}) " +
        log"because some of its tasks had failed: " +
        log"${MDC(PARTITION_IDS, shuffleStage.findMissingPartitions().mkString(", "))}")
      submitStage(shuffleStage) // resubmit without check
    } else {
      markMapStageJobsAsFinished(shuffleStage)
      submitWaitingChildStages(shuffleStage)
    } 
}