[
https://issues.apache.org/jira/browse/SPARK-29582?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel
]
Hyukjin Kwon reassigned SPARK-29582:
------------------------------------
Assignee: Xianyang Liu
> Unify the behavior of pyspark.TaskContext with spark core
> ---------------------------------------------------------
>
> Key: SPARK-29582
> URL: https://issues.apache.org/jira/browse/SPARK-29582
> Project: Spark
> Issue Type: Bug
> Components: PySpark
> Affects Versions: 2.4.4
> Reporter: Xianyang Liu
> Assignee: Xianyang Liu
> Priority: Major
>
> In Spark Core, there is a `TaskContext` object which is a singleton. We set a
> task context instance which can be TaskContext or BarrierTaskContext before
> the task function startup, and unset it to none after the function end. So we
> can both get TaskContext and BarrierTaskContext with the object. How we can
> only get the BarrierTaskContext with `BarrierTaskContext`, we will get `None`
> if we get it by `TaskContext.get` in a barrier stage.
>
> In this patch, we unify the behavior of TaskContext for pyspark with Spark
> core. This is useful when people switch from normal code to barrier code, and
> only need a little update.
--
This message was sent by Atlassian Jira
(v8.3.4#803005)
---------------------------------------------------------------------
To unsubscribe, e-mail: issues-unsubscr...@spark.apache.org
For additional commands, e-mail: issues-h...@spark.apache.org
