Description
Currently the TaskContext info isn't exposed in PySpark. While we don't need to expose the full TaskContext information in Python it would make sense to expose the public APIs in Python for users who are doing custom logging or job handling with task id or retry attempt.