[
https://issues.apache.org/jira/browse/SPARK-42235?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=17687447#comment-17687447
]
Apache Spark commented on SPARK-42235:
--------------------------------------
User 'wayneguow' has created a pull request for this issue:
https://github.com/apache/spark/pull/39974
> Missing typing for pandas_udf
> -----------------------------
>
> Key: SPARK-42235
> URL: https://issues.apache.org/jira/browse/SPARK-42235
> Project: Spark
> Issue Type: Bug
> Components: PySpark
> Affects Versions: 3.3.0
> Reporter: Donggu Kang
> Priority: Minor
>
> The typing stub {{site-packages/pyspark/sql/pandas/functions.pyi}} has a list
> of possible signatures of {{pandas_udf}}. It is missing a case
> {code:python}
> import pyspark.sql.functions as F
> # PySpark3's typing stub error
> @F.pandas_udf(
> returnType=LongType()
> )
> def my_udf(dummy_col: pd.Series) -> pd.Series:
> ...
> {code}
> The stub defined {{pandas_udf(f, returnType, functionType)}} but not
> {{pandas_udf(f, returnType)}}. The official documentation recommends using a
> return type hint instead of {{returnType}}.
>
> {code:python}
> # defined
> @overload
> def pandas_udf( f: PandasScalarToScalarFunction, returnType:
> Union[AtomicDataTypeOrString, ArrayType], functionType: PandasScalarUDFType,
> ) -> UserDefinedFunctionLike: ...
>
> # not defined
> @overload
> def pandas_udf( f: PandasScalarToScalarFunction, returnType:
> Union[AtomicDataTypeOrString, ArrayType]) -> UserDefinedFunctionLike: ...
> {code}
>
--
This message was sent by Atlassian Jira
(v8.20.10#820010)
---------------------------------------------------------------------
To unsubscribe, e-mail: issues-unsubscr...@spark.apache.org
For additional commands, e-mail: issues-h...@spark.apache.org
