pyspark.TaskContext.get#

classmethod TaskContext.get()[source]#

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

Returns:

TaskContext, optional

Notes

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