[jira] [Created] (SPARK-55059) Remove empty table workaround in toPandas

["Yicong Huang (Jira)" <ji...@apache.org>]

Yicong Huang created SPARK-55059:
------------------------------------

             Summary: Remove empty table workaround in toPandas
                 Key: SPARK-55059
                 URL: https://issues.apache.org/jira/browse/SPARK-55059
             Project: Spark
          Issue Type: Improvement
          Components: PySpark
    Affects Versions: 4.2.0
            Reporter: Yicong Huang


SPARK-51112 added a workaround in \{{_convert_arrow_table_to_pandas()}} to avoid segfault when converting empty tables with nested array columns:

{code:python}
# SPARK-51112: If the table is empty, we avoid using pyarrow to_pandas to create the
# DataFrame, as it may fail with a segmentation fault.
if arrow_table.num_rows == 0:
    column_data = (
        pd.Series([], name=temp_col_names[i], dtype="object") for i in range(len(schema.fields))
    )
{code}

This workaround is no longer necessary after SPARK-55056, which fixed the root cause in \{{ArrayWriter.finish()}} by properly initializing the Arrow ListArray offset buffer when \{{count == 0}}.

Proposal: Remove the SPARK-51112 workaround and let pyarrow handle empty tables directly.



--
This message was sent by Atlassian Jira
(v8.20.10#820010)

---------------------------------------------------------------------
To unsubscribe, e-mail: issues-unsubscribe@spark.apache.org
For additional commands, e-mail: issues-help@spark.apache.org