[ 
https://issues.apache.org/jira/browse/SPARK-28153?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel
 ]

Dongjoon Hyun updated SPARK-28153:
----------------------------------
    Affects Version/s: 2.4.3

> input_file_name doesn't work with Python UDF in the same project
> ----------------------------------------------------------------
>
>                 Key: SPARK-28153
>                 URL: https://issues.apache.org/jira/browse/SPARK-28153
>             Project: Spark
>          Issue Type: Bug
>          Components: PySpark
>    Affects Versions: 3.0.0, 2.4.3
>            Reporter: Hyukjin Kwon
>            Assignee: Hyukjin Kwon
>            Priority: Major
>             Fix For: 3.0.0
>
>
> {code}
> from pyspark.sql.functions import udf, input_file_name
> spark.range(10).write.mode("overwrite").parquet("/tmp/foo")
> spark.read.parquet("/tmp/foo").select(udf(lambda x: x, "long")("id"), 
> input_file_name()).show()
> {code}
> {code}
> +------------+-----------------+
> |<lambda>(id)|input_file_name()|
> +------------+-----------------+
> |           8|                 |
> |           5|                 |
> |           0|                 |
> |           9|                 |
> |           6|                 |
> |           2|                 |
> |           3|                 |
> |           4|                 |
> |           7|                 |
> |           1|                 |
> +------------+-----------------+
> {code}



--
This message was sent by Atlassian JIRA
(v7.6.14#76016)

---------------------------------------------------------------------
To unsubscribe, e-mail: issues-unsubscr...@spark.apache.org
For additional commands, e-mail: issues-h...@spark.apache.org

Reply via email to