pyspark.TaskContext.attemptNumber

TaskContext.attemptNumber() → int

” How many times this task has been attempted. The first task attempt will be assigned attemptNumber = 0, and subsequent attempts will have increasing attempt numbers.