Github user icexelloss commented on a diff in the pull request:
https://github.com/apache/spark/pull/21383#discussion_r191469832
--- 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 --
Can we also add tests for pandas_udf?
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
