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: reviews-unsubscr...@spark.apache.org For additional commands, e-mail: reviews-h...@spark.apache.org