From 869d4181264d03ddfb3006e73eadb84d409d72bf Mon Sep 17 00:00:00 2001 From: Ruifeng Zheng Date: Thu, 7 Aug 2025 15:41:21 +0800 Subject: [PATCH] xyz --- python/pyspark/sql/tests/arrow/test_arrow_udf_scalar.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/pyspark/sql/tests/arrow/test_arrow_udf_scalar.py b/python/pyspark/sql/tests/arrow/test_arrow_udf_scalar.py index 97db29ae6590..15e5123b65ed 100644 --- a/python/pyspark/sql/tests/arrow/test_arrow_udf_scalar.py +++ b/python/pyspark/sql/tests/arrow/test_arrow_udf_scalar.py @@ -97,7 +97,7 @@ def test_arrow_udf_output_nested_arrays(self): df = self.spark.createDataFrame([("hi boo",), ("bye boo",)], ["vals"]) tokenize = arrow_udf( - lambda s: pa.array([pa.compute.ascii_split_whitespace(s).to_pylist()]), + lambda s: pa.array([[v] for v in pa.compute.ascii_split_whitespace(s).to_pylist()]), ArrayType(ArrayType(StringType())), )