[
https://issues.apache.org/jira/browse/SPARK-34799?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel
]
Darcy Shen updated SPARK-34799:
-------------------------------
Summary: Return UDT from Pandas UDF: @pandas_udf(UDT) (was: Return
User-defined types from Pandas UDF)
> Return UDT from Pandas UDF: @pandas_udf(UDT)
> --------------------------------------------
>
> Key: SPARK-34799
> URL: https://issues.apache.org/jira/browse/SPARK-34799
> Project: Spark
> Issue Type: Sub-task
> Components: PySpark, SQL
> Affects Versions: 3.0.2, 3.1.1
> Reporter: Darcy Shen
> Priority: Major
>
> Because Pandas UDF uses pyarrow to passing data, it does not currently
> support UserDefinedTypes, as what normal python udf does.
> For example:
> {code:python}
> class BoxType(UserDefinedType):
> @classmethod
> def sqlType(cls) -> StructType:
> return StructType(
> fields=[
> StructField("xmin", DoubleType(), False),
> StructField("ymin", DoubleType(), False),
> StructField("xmax", DoubleType(), False),
> StructField("ymax", DoubleType(), False),
> ]
> )
> @pandas_udf(
> returnType=StructType([StructField("boxes", ArrayType(Box()))]
> )
> def pandas_pf(s: pd.DataFrame) -> pd.DataFrame:
> yield s
> {code}
> The logs show
> {code}
> try:
> to_arrow_type(self._returnType_placeholder)
> except TypeError:
> > raise NotImplementedError(
> "Invalid return type with scalar Pandas UDFs: %s is "
> E NotImplementedError: Invalid return type with scalar
> Pandas UDFs: StructType(List(StructField(boxes,ArrayType(Box,true),true))) is
> not supported
> {code}
--
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
