pyspark.BarrierTaskContext.partitionId

BarrierTaskContext.partitionId()

The ID of the RDD partition that is computed by this task.