Skip tests that hang on DB9.1

NVIDIA · Jan 21, 2022 · b4ceb5a · b4ceb5a
1 parent 90584ba
commit b4ceb5a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/integration_tests/src/main/python/udf_test.py b/integration_tests/src/main/python/udf_test.py
@@ -15,6 +15,7 @@
 import pytest
 
 from conftest import is_at_least_precommit_run
+from spark_session import is_databricks91_or_later
 
 from pyspark.sql.pandas.utils import require_minimum_pyarrow_version, require_minimum_pandas_version
 try:
@@ -212,6 +213,7 @@ def pandas_sum(to_process: pd.Series) -> list:
 
 
 # ======= Test flat map group in Pandas =======
+@pytest.mark.skipif(is_databricks91_or_later(), reason="https://github.com/NVIDIA/spark-rapids/issues/4599")
 @ignore_order
 @pytest.mark.parametrize('data_gen', [LongGen()], ids=idfn)
 def test_group_apply_udf(data_gen):
@@ -226,6 +228,7 @@ def pandas_add(data):
             conf=arrow_udf_conf)
 
 
+@pytest.mark.skipif(is_databricks91_or_later(), reason="https://github.com/NVIDIA/spark-rapids/issues/4599")
 @ignore_order
 @pytest.mark.parametrize('data_gen', arrow_common_gen, ids=idfn)
 def test_group_apply_udf_more_types(data_gen):