From 1c0d9744e408a3284649cbce8ae3a8ab7fce347c Mon Sep 17 00:00:00 2001 From: rjzamora Date: Tue, 17 Dec 2024 14:23:43 -0800 Subject: [PATCH] trigger explicit-comms-batchsize error --- dask_cuda/tests/test_explicit_comms.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dask_cuda/tests/test_explicit_comms.py b/dask_cuda/tests/test_explicit_comms.py index 96ff91e9..2f79251d 100644 --- a/dask_cuda/tests/test_explicit_comms.py +++ b/dask_cuda/tests/test_explicit_comms.py @@ -239,7 +239,7 @@ def check_shuffle(): ): dask.config.refresh() # Trigger re-read of the environment variables with pytest.raises(ValueError, match="explicit-comms-batchsize"): - ddf.shuffle(on="key", npartitions=4) + ddf.shuffle(on="key", npartitions=4).dask if in_cluster: with LocalCluster(