trigger explicit-comms-batchsize error

rapidsai · Dec 17, 2024 · 1c0d974 · 1c0d974
1 parent 32bcb22
commit 1c0d974
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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(