From 5fda55a3510ae9c0a29359a12deff97f03e2c4f6 Mon Sep 17 00:00:00 2001 From: "Robert (Bobby) Evans" Date: Fri, 13 Dec 2024 09:28:09 -0600 Subject: [PATCH] Update to deal with bools off for ORC --- integration_tests/src/main/python/orc_write_test.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/integration_tests/src/main/python/orc_write_test.py b/integration_tests/src/main/python/orc_write_test.py index 672d6c44e8a..103cae474a3 100644 --- a/integration_tests/src/main/python/orc_write_test.py +++ b/integration_tests/src/main/python/orc_write_test.py @@ -151,7 +151,11 @@ def test_write_round_trip_nullable_struct(spark_tmp_path, gen, orc_impl): lambda spark, path: unary_op_df(spark, gen_for_struct, num_slices=1).write.orc(path), lambda spark, path: spark.read.orc(path), data_path, - conf={'spark.sql.orc.impl': orc_impl, 'spark.rapids.sql.format.orc.write.enabled': True}) + conf={'spark.sql.orc.impl': orc_impl, + 'spark.rapids.sql.format.orc.write.enabled': True, + # https://github.com/NVIDIA/spark-rapids/issues/11736, so verify that we still do it correctly + # once this is fixed + 'spark.rapids.sql.format.orc.write.boolType.enabled' : True}) orc_part_write_gens = [ # Add back boolean_gen when https://github.com/rapidsai/cudf/issues/6763 is fixed