From 4ba0bcf4b1284f0051324dd85f6ce1c1bef3c6af Mon Sep 17 00:00:00 2001 From: Parth Chandra Date: Fri, 20 Mar 2026 15:33:05 -0700 Subject: [PATCH] fix: disable ANSI mode in benchmarks to avoid exceptions on invalid input comet-test-apache-spark defaults spark.sql.ansi.enabled to true, causing CAST on intentionally invalid benchmark data to throw instead of returning NULL. Co-Authored-By: Claude Sonnet 4.6 --- .../org/apache/spark/sql/benchmark/CometBenchmarkBase.scala | 2 ++ 1 file changed, 2 insertions(+) diff --git a/spark/src/test/scala/org/apache/spark/sql/benchmark/CometBenchmarkBase.scala b/spark/src/test/scala/org/apache/spark/sql/benchmark/CometBenchmarkBase.scala index 2a81316c95..deade5e337 100644 --- a/spark/src/test/scala/org/apache/spark/sql/benchmark/CometBenchmarkBase.scala +++ b/spark/src/test/scala/org/apache/spark/sql/benchmark/CometBenchmarkBase.scala @@ -63,6 +63,8 @@ trait CometBenchmarkBase sparkSession.conf.set(SQLConf.WHOLESTAGE_CODEGEN_ENABLED.key, "true") sparkSession.conf.set(CometConf.COMET_ENABLED.key, "false") sparkSession.conf.set(CometConf.COMET_EXEC_ENABLED.key, "false") + // Benchmarks use invalid input values that should produce NULL, not exceptions + sparkSession.conf.set(SQLConf.ANSI_ENABLED.key, "false") sparkSession }