diff --git a/dask_cuda/tests/test_explicit_comms.py b/dask_cuda/tests/test_explicit_comms.py index 05edbfb8..dbff3b92 100644 --- a/dask_cuda/tests/test_explicit_comms.py +++ b/dask_cuda/tests/test_explicit_comms.py @@ -99,6 +99,7 @@ def _test_dataframe_merge(backend, protocol, n_workers): @pytest.mark.parametrize("nworkers", [1, 2, 4]) @pytest.mark.parametrize("backend", ["pandas", "cudf"]) @pytest.mark.parametrize("protocol", ["tcp", "ucx"]) +@pytest.mark.xfail(reason="https://github.com/rapidsai/dask-cuda/issues/549") def test_dataframe_merge(backend, protocol, nworkers): if backend == "cudf": pytest.importorskip("cudf") @@ -203,6 +204,7 @@ def _test_dataframe_shuffle(backend, protocol, n_workers): @pytest.mark.parametrize("nworkers", [1, 2, 3]) @pytest.mark.parametrize("backend", ["pandas", "cudf"]) @pytest.mark.parametrize("protocol", ["tcp", "ucx"]) +@pytest.mark.xfail(reason="https://github.com/rapidsai/dask-cuda/issues/549") def test_dataframe_shuffle(backend, protocol, nworkers): if backend == "cudf": pytest.importorskip("cudf") @@ -243,6 +245,7 @@ def check_shuffle(in_cluster): check_shuffle(False) +@pytest.mark.xfail(reason="https://github.com/rapidsai/dask-cuda/issues/549") def test_dask_use_explicit_comms(): p = mp.Process(target=_test_dask_use_explicit_comms) p.start()