Restore hash aggregate tests after cub segmented sort fix [databricks] (

#7142) * Revert "Skip test_hash_groupby_collect_with_single_distinct (#7107)" This reverts commit d5ba2e2. Signed-off-by: Jason Lowe <[email protected]> * Revert "Skip test_hash_groupby_collect_partial_replace_with_distinct_fallback for #7092 [databricks] (#7102)" This reverts commit c70bcfb. Signed-off-by: Jason Lowe <[email protected]> Signed-off-by: Jason Lowe <[email protected]>
NVIDIA · Nov 23, 2022 · 2004768 · 2004768
1 parent b7ada9f
commit 2004768
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/integration_tests/src/main/python/hash_aggregate_test.py b/integration_tests/src/main/python/hash_aggregate_test.py
@@ -731,7 +731,6 @@ def do_it(spark):
 
 @ignore_order(local=True)
 @pytest.mark.parametrize('data_gen', _full_gen_data_for_collect_op, ids=idfn)
-@pytest.mark.skip(reason='https://github.com/NVIDIA/spark-rapids/issues/7092,7104')
 def test_hash_groupby_collect_with_single_distinct(data_gen):
     # test collect_ops with other distinct aggregations
     assert_gpu_and_cpu_are_equal_collect(
@@ -840,7 +839,6 @@ def test_hash_groupby_collect_partial_replace_fallback(data_gen,
 @pytest.mark.parametrize('aqe_enabled', ['false', 'true'], ids=idfn)
 @pytest.mark.parametrize('use_obj_hash_agg', ['false', 'true'], ids=idfn)
 @pytest.mark.xfail(condition=is_databricks104_or_later(), reason='https://github.com/NVIDIA/spark-rapids/issues/4963')
-@pytest.mark.skip(reason='https://github.com/NVIDIA/spark-rapids/issues/7092')
 def test_hash_groupby_collect_partial_replace_with_distinct_fallback(data_gen,
                                                                      replace_mode,
                                                                      aqe_enabled,