Hyukjin Kwon created SPARK-60962:
------------------------------------
Summary: Pickle user-defined types of __main__ as UDF return types
in Spark Connect Rust as PySpark does
Key: SPARK-60962
URL: https://issues.apache.org/jira/browse/SPARK-60962
Project: Spark
Issue Type: Bug
Components: Connect
Affects Versions: connect-rust-4.3.0
Reporter: Hyukjin Kwon
In PySpark, a user-defined type that a notebook or script defines, in
{{__main__}}, can be the return type of a UDF: the UDT instance is pickled with
cloudpickle, which sends a {{__main__}} class by value. The Rust-backed Python
client pickles a UDT as its JSON, which the worker rebuilds with
{{UserDefinedType.fromJson}}: that imports the class by its module, which
{{__main__}} on the worker does not define, so {{udf(f,
returnType=PointUDT())}} fails on the worker with {{UNSUPPORTED_OPERATION}},
and a UDT whose {{__init__}} takes arguments or keeps state is rebuilt without
them. Pickle a UDT instance as PySpark does.
--
This message was sent by Atlassian Jira
(v10.3.26#10030026)
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
