pyspark.TaskContext.stageId

TaskContext.stageId() → int

The ID of the stage that this task belong to.

Returns

int

current stage id.

pyspark.TaskContext.resources pyspark.TaskContext.taskAttemptId