Same here; doing these in DAGScheduler.getMissingParentStages may be too late, as that happens in response to Akka events, and will lead to a weirder error.