Update test_udtf.py

AngersZhuuuu · Jun 24, 2024 · ea1f9f1 · ea1f9f1
1 parent 52e6c91
commit ea1f9f1
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
@@ -2820,7 +2820,7 @@ def eval(self):
                 yield ExamplePoint(0, 1),
 
         schema = StructType().add("point", ExamplePointUDT())
-        [row] =udtf(TestUDTF, returnType=schema, useArrow=True)(lit()).collect()
+        [row] = udtf(TestUDTF, returnType=schema, useArrow=True)().collect()
         self.assertEqual(row[0], ExamplePoint(1.0, 2.0))