pyspark.TaskContext.get

classmethod TaskContext.get() → Optional[pyspark.taskcontext.TaskContext]

Return the currently active TaskContext. This can be called inside of user functions to access contextual information about running tasks.

Notes

Must be called on the worker, not the driver. Returns None if not initialized.