From 52d89f5b624d64d3b41537c790c771cd8336d28b Mon Sep 17 00:00:00 2001 From: Angerszhuuuu Date: Mon, 24 Jun 2024 18:07:22 +0800 Subject: [PATCH] Update test_udtf.py --- python/pyspark/sql/tests/test_udtf.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/pyspark/sql/tests/test_udtf.py b/python/pyspark/sql/tests/test_udtf.py index 12a6faa8ca1a1..679339cb0ebf8 100644 --- a/python/pyspark/sql/tests/test_udtf.py +++ b/python/pyspark/sql/tests/test_udtf.py @@ -2820,10 +2820,10 @@ def eval(self): yield ExamplePoint(0, 1), schema = StructType().add("point", ExamplePointUDT()) - [row] = udtf(TestUDTF, returnType=schema, useArrow=False).collect() + [row] = udtf(TestUDTF, returnType=schema, useArrow=False)().collect() self.assertEqual(row[0], ExamplePoint(1.0, 2.0)) with self.assertRaisesRegex(PythonException, "UDTF_ARROW_TYPE_CAST_ERROR"): - udtf(TestUDTF, returnType=schema, useArrow=True).collect() + udtf(TestUDTF, returnType=schema, useArrow=True)().collect() class UDTFArrowTests(UDTFArrowTestsMixin, ReusedSQLTestCase):