pyspark.TaskContext.stageId

TaskContext.stageId()[source]

The ID of the stage that this task belong to.