Skip zero-sized table and non-None partition info

Seems to induce a bug in libcudf for now.
rapidsai · Nov 19, 2024 · f2a905e · f2a905e
1 parent 91e847e
commit f2a905e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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()])