Github user HyukjinKwon commented on a diff in the pull request:
https://github.com/apache/spark/pull/21383#discussion_r190666471
--- Diff: python/pyspark/sql/udf.py ---
@@ -92,7 +92,7 @@ def __init__(self, func,
raise TypeError(
"Invalid evalType: evalType should be an int but is
{}".format(evalType))
- self.func = func
+ self.func = fail_on_stopiteration(func)
--- End diff --
This is because of this from a very quick look. This looks __doc__. You
should wrap it with `@functools.wraps`.
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
