Kaicheng Zhou created SPARK-59233:
-------------------------------------

             Summary: Pass the session's Python worker environment to every 
Python function family
                 Key: SPARK-59233
                 URL: https://issues.apache.org/jira/browse/SPARK-59233
             Project: Spark
          Issue Type: Task
          Components: PySpark, Spark Core
    Affects Versions: 5.0.0, 4.4.0
            Reporter: Kaicheng Zhou


SPARK-58752 added the {{spark.pythonWorkerEnv.*}} session configurations, which 
set environment variables for the Python workers that run a session's Python 
functions. It
installed them for the regular scalar Python UDF only, so the same logic 
written as a pandas UDF, a UDTF or {{applyInPandas}} received nothing — a 
difference users cannot
predict from anything in the API.

Extend the environment to every Python worker a session launches for a Python 
function that session supplied: scalar UDFs in all serialization modes, 
{{mapInPandas}} and
{{{}mapInArrow{}}}, grouped-map, cogrouped-map, grouped-aggregate and window 
functions, row and Arrow UDTFs, {{applyInPandasWithState}} and 
{{{}transformWithState{}}}, Python data sources including the planning and 
streaming-reader workers, and the {{foreachBatch}} worker.



--
This message was sent by Atlassian Jira
(v8.20.10#820010)

---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]

Reply via email to