Github user ueshin commented on a diff in the pull request:
https://github.com/apache/spark/pull/20288#discussion_r161969687
--- Diff: python/pyspark/sql/context.py ---
@@ -147,7 +147,8 @@ def udf(self):
:return: :class:`UDFRegistration`
"""
- return UDFRegistration(self)
+ from pyspark.sql.session import UDFRegistration
+ return UDFRegistration(self.sparkSession)
--- End diff --
How about `return self.sparkSession.udf`?
---
---------------------------------------------------------------------
To unsubscribe, e-mail: reviews-unsubscr...@spark.apache.org
For additional commands, e-mail: reviews-h...@spark.apache.org
