diff --git a/dask_cuda/tests/test_explicit_comms.py b/dask_cuda/tests/test_explicit_comms.py index b39f2f2b..25a2b257 100644 --- a/dask_cuda/tests/test_explicit_comms.py +++ b/dask_cuda/tests/test_explicit_comms.py @@ -25,10 +25,11 @@ mp = mp.get_context("spawn") # type: ignore ucp = pytest.importorskip("ucp") +QUERY_PLANNING_ON = dask.config.get("dataframe.query-planning", None) is not False # Skip these tests when dask-expr is active (for now) query_planning_skip = pytest.mark.skipif( - dask.config.get("dataframe.query-planning", None) is not False, + QUERY_PLANNING_ON, reason=( "The 'explicit-comms' config is not supported " "when query planning is enabled." @@ -185,6 +186,10 @@ def test_dataframe_shuffle(backend, protocol, nworkers, _partitions): if backend == "cudf": pytest.importorskip("cudf") + if QUERY_PLANNING_ON: + # There seem to be problems with dask-expr + dask<2024.2.1 + pytest.importorskip("dask", minversion="2024.2.1") + p = mp.Process( target=_test_dataframe_shuffle, args=(backend, protocol, nworkers, _partitions) )