From 8871c74a7f6d2a48d7d5fb037b77ea0e4b9afa92 Mon Sep 17 00:00:00 2001 From: James Bourbeau Date: Wed, 14 Sep 2022 12:32:16 -0500 Subject: [PATCH 1/2] Avoid passing groupby key list of length one --- dask/dataframe/io/parquet/arrow.py | 1 + 1 file changed, 1 insertion(+) diff --git a/dask/dataframe/io/parquet/arrow.py b/dask/dataframe/io/parquet/arrow.py index b9799e25082..86327ec2d5a 100644 --- a/dask/dataframe/io/parquet/arrow.py +++ b/dask/dataframe/io/parquet/arrow.py @@ -111,6 +111,7 @@ def _write_partitioned( subschema = subschema.remove(subschema.get_field_index(col)) md_list = [] + partition_keys = partition_keys[0] if len(partition_keys) == 1 else partition_keys for keys, subgroup in data_df.groupby(partition_keys): if not isinstance(keys, tuple): keys = (keys,) From 980703234a8e610976f76da0f83e54f6a0b162ae Mon Sep 17 00:00:00 2001 From: James Bourbeau Date: Wed, 14 Sep 2022 12:32:31 -0500 Subject: [PATCH 2/2] test-upstream