Github user e-dorigatti commented on a diff in the pull request: https://github.com/apache/spark/pull/21467#discussion_r192116000 --- Diff: python/pyspark/worker.py --- @@ -69,6 +69,7 @@ def chain(f, g): def wrap_udf(f, return_type): + f = fail_on_stopiteration(f) --- End diff -- Or just move the call to `get_argspec` in `read_single_udf`
--- --------------------------------------------------------------------- To unsubscribe, e-mail: reviews-unsubscr...@spark.apache.org For additional commands, e-mail: reviews-h...@spark.apache.org