Github user icexelloss commented on a diff in the pull request:
https://github.com/apache/spark/pull/21383#discussion_r191470695
--- Diff: python/pyspark/sql/tests.py ---
@@ -900,6 +900,22 @@ def __call__(self, x):
self.assertEqual(f, f_.func)
self.assertEqual(return_type, f_.returnType)
+ def test_stopiteration_in_udf(self):
--- End diff --
We can also merge this PR first. I can follow up with the pandas udf tests.
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
