Fix arrow UDTF support by switching to a new exception class

dtenedor · Nov 7, 2023 · 44a8438 · 44a8438
1 parent 1b3fc2d
commit 44a8438
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/pyspark/sql/tests/test_udtf.py b/python/pyspark/sql/tests/test_udtf.py
@@ -2483,7 +2483,7 @@ def analyze(_):
             def eval(self, _: Row):
                 self._total += 1
                 if self._total >= 4:
-                    raise SkipRestOfInputTableException("StopIteration at self._total >= 4")
+                    raise SkipRestOfInputTableException("Stop at self._total >= 4")
 
             def terminate(self):
                 yield self._total,