From f2a905eb82bde1d6fb27301cafed7635cff5ec15 Mon Sep 17 00:00:00 2001 From: Lawrence Mitchell Date: Tue, 19 Nov 2024 14:55:58 +0000 Subject: [PATCH] Skip zero-sized table and non-None partition info Seems to induce a bug in libcudf for now. --- python/pylibcudf/pylibcudf/tests/io/test_parquet.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/python/pylibcudf/pylibcudf/tests/io/test_parquet.py b/python/pylibcudf/pylibcudf/tests/io/test_parquet.py index 875b6cef264..0b368428f7b 100644 --- a/python/pylibcudf/pylibcudf/tests/io/test_parquet.py +++ b/python/pylibcudf/pylibcudf/tests/io/test_parquet.py @@ -162,6 +162,8 @@ def test_write_parquet( max_dictionary_size, ): _, pa_table = table_data + if len(pa_table) == 0 and partitions is not None: + pytest.skip("https://github.com/rapidsai/cudf/issues/17361") plc_table = plc.interop.from_arrow(pa_table) table_meta = plc.io.types.TableInputMetadata(plc_table) sink = plc.io.SinkInfo([io.BytesIO()])