From 8171fdad3ba6fdbaa1bc6e41159ce1074e916c0c Mon Sep 17 00:00:00 2001 From: Jim Brennan Date: Wed, 3 Jan 2024 12:57:39 -0600 Subject: [PATCH] fix test_hash_agg_with_nan_keys floating point sum failure Signed-off-by: Jim Brennan --- integration_tests/src/main/python/hash_aggregate_test.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/integration_tests/src/main/python/hash_aggregate_test.py b/integration_tests/src/main/python/hash_aggregate_test.py index 3fe7bae2a28..55e1ca44c6f 100644 --- a/integration_tests/src/main/python/hash_aggregate_test.py +++ b/integration_tests/src/main/python/hash_aggregate_test.py @@ -159,7 +159,7 @@ _grpkey_doubles_with_nan_zero_grouping_keys = [ ('a', RepeatSeqGen(DoubleGen(nullable=(True, 10.0), special_cases=_nan_zero_double_special_cases), length=50)), - ('b', FloatGen(nullable=(True, 10.0))), + ('b', IntegerGen(nullable=(True, 10.0))), ('c', LongGen())] # Schema for xfail cases @@ -1154,7 +1154,6 @@ def test_hash_multiple_filters(data_gen, conf): 'min(a), max(b) filter (where c > 250) from hash_agg_table group by a', conf) -@datagen_overrides(seed=0, reason='https://github.com/NVIDIA/spark-rapids/issues/10026') @approximate_float @ignore_order @pytest.mark.parametrize('data_gen', [_grpkey_floats_with_nan_zero_grouping_keys, @@ -1222,7 +1221,7 @@ def test_hash_agg_with_struct_of_array_fallback(data_gen): @approximate_float @ignore_order -@pytest.mark.parametrize('data_gen', [ _grpkey_doubles_with_nan_zero_grouping_keys], ids=idfn) +@pytest.mark.parametrize('data_gen', [ _grpkey_floats_with_nulls_and_nans ], ids=idfn) def test_count_distinct_with_nan_floats(data_gen): assert_gpu_and_cpu_are_equal_sql( lambda spark : gen_df(spark, data_gen, length=1024),