diff --git a/integration_tests/src/main/python/udf_cudf_test.py b/integration_tests/src/main/python/udf_cudf_test.py index e06984a4993..6385422c437 100644 --- a/integration_tests/src/main/python/udf_cudf_test.py +++ b/integration_tests/src/main/python/udf_cudf_test.py @@ -124,6 +124,7 @@ def gpu_run(spark): _assert_cpu_gpu(cpu_run, gpu_run, gpu_conf=_conf) +@pytest.mark.skip("https://github.com/NVIDIA/spark-rapids/issues/746") @allow_non_gpu('GpuMapInPandasExec','PythonUDF') @cudf_udf def test_map_in_pandas(): @@ -160,6 +161,7 @@ def _normalize_gpu_func(df): v = gdf.v return gdf.assign(v=(v - v.mean()) / v.std()).to_pandas() +@pytest.mark.skip("https://github.com/NVIDIA/spark-rapids/issues/746") @allow_non_gpu('GpuFlatMapGroupsInPandasExec','PythonUDF') @cudf_udf def test_group_apply(): @@ -174,6 +176,7 @@ def gpu_run(spark): _assert_cpu_gpu(cpu_run, gpu_run, gpu_conf=_conf, is_sort=True) +@pytest.mark.skip("https://github.com/NVIDIA/spark-rapids/issues/746") @allow_non_gpu('GpuFlatMapGroupsInPandasExec','PythonUDF') @cudf_udf def test_group_apply_in_pandas(): @@ -206,6 +209,7 @@ def _sum_gpu_func(v: pd.Series) -> int: gpu_serises = cudf.Series(v) return gpu_serises.sum() +@pytest.mark.skip("https://github.com/NVIDIA/spark-rapids/issues/746") @allow_non_gpu('GpuAggregateInPandasExec','PythonUDF','Alias') @cudf_udf def test_group_agg(): @@ -220,6 +224,7 @@ def gpu_run(spark): _assert_cpu_gpu(cpu_run, gpu_run, gpu_conf=_conf, is_sort=True) +@pytest.mark.skip("https://github.com/NVIDIA/spark-rapids/issues/746") @allow_non_gpu('GpuAggregateInPandasExec','PythonUDF','Alias') @cudf_udf def test_sql_group(): @@ -236,6 +241,7 @@ def gpu_run(spark): _assert_cpu_gpu(cpu_run, gpu_run, gpu_conf=_conf, is_sort=True) +@pytest.mark.skip("https://github.com/NVIDIA/spark-rapids/issues/746") @allow_non_gpu('GpuWindowInPandasExec','PythonUDF','Alias','WindowExpression','WindowSpecDefinition','SpecifiedWindowFrame','UnboundedPreceding$', 'UnboundedFollowing$') @cudf_udf def test_window(): @@ -252,6 +258,7 @@ def gpu_run(spark): _assert_cpu_gpu(cpu_run, gpu_run, gpu_conf=_conf, is_sort=True) +@pytest.mark.skip("https://github.com/NVIDIA/spark-rapids/issues/746") @allow_non_gpu('GpuFlatMapCoGroupsInPandasExec','PythonUDF') @cudf_udf def test_cogroup(): diff --git a/integration_tests/src/main/python/udf_test.py b/integration_tests/src/main/python/udf_test.py index 276fe15d8d6..7cdc46ee1b5 100644 --- a/integration_tests/src/main/python/udf_test.py +++ b/integration_tests/src/main/python/udf_test.py @@ -88,6 +88,7 @@ def pandas_sum(to_process: pd.Series) -> int: .agg(pandas_sum(f.col('b'))), conf=arrow_udf_conf) +@pytest.mark.skip("https://github.com/NVIDIA/spark-rapids/issues/740") @ignore_order @allow_non_gpu('WindowInPandasExec', 'PythonUDF', 'WindowExpression', 'Alias', 'WindowSpecDefinition', 'SpecifiedWindowFrame', 'UnboundedPreceding$', 'UnboundedFollowing$') @pytest.mark.parametrize('data_gen', integral_gens, ids=idfn)