diff --git a/integration_tests/src/main/python/date_time_test.py b/integration_tests/src/main/python/date_time_test.py index 78586197d84..69ec80a17c7 100644 --- a/integration_tests/src/main/python/date_time_test.py +++ b/integration_tests/src/main/python/date_time_test.py @@ -456,21 +456,17 @@ def test_date_format_for_date(data_gen, date_format): @pytest.mark.parametrize('data_gen', [timestamp_gen], ids=idfn) @pytest.mark.skipif(not is_supported_time_zone(), reason="not all time zones are supported now, refer to https://github.com/NVIDIA/spark-rapids/issues/6839, please update after all time zones are supported") def test_date_format_for_time(data_gen, date_format): - conf = {'spark.rapids.sql.nonUTC.enabled': True} assert_gpu_and_cpu_are_equal_collect( - lambda spark : unary_op_df(spark, data_gen).selectExpr("date_format(a, '{}')".format(date_format)), - conf) + lambda spark : unary_op_df(spark, data_gen).selectExpr("date_format(a, '{}')".format(date_format))) @pytest.mark.parametrize('date_format', supported_date_formats, ids=idfn) @pytest.mark.parametrize('data_gen', [timestamp_gen], ids=idfn) @pytest.mark.skipif(is_supported_time_zone(), reason="not all time zones are supported now, refer to https://github.com/NVIDIA/spark-rapids/issues/6839, please update after all time zones are supported") @allow_non_gpu('ProjectExec') def test_date_format_for_time_fall_back(data_gen, date_format): - conf = {'spark.rapids.sql.nonUTC.enabled': True} assert_gpu_fallback_collect( lambda spark : unary_op_df(spark, data_gen).selectExpr("date_format(a, '{}')".format(date_format)), - 'ProjectExec', - conf) + 'ProjectExec') @pytest.mark.parametrize('date_format', supported_date_formats + ['yyyyMMdd'], ids=idfn) # from 0001-02-01 to 9999-12-30 to avoid 'year 0 is out of range'