Add fix for test_groupby_maintain_order_random

rapidsai · Feb 21, 2025 · 0558235 · 0558235
1 parent 3adbc9f
commit 0558235
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/python/cudf_polars/tests/conftest.py b/python/cudf_polars/tests/conftest.py
@@ -47,8 +47,13 @@ def pytest_sessionstart(session):
         session.config.getoption("--dask-cluster")
         and session.config.getoption("--executor") == "dask-experimental"
     ):
+        from dask import config
         from dask.distributed import Client, LocalCluster
 
+        # Avoid "Sending large graph of size ..." warnings
+        # (We expect these for tests using literal/random arrays)
+        config.set({"distributed.admin.large-graph-warning-threshold": "20MB"})
+
         cluster = LocalCluster()
         client = Client(cluster)
         session.stash[DISTRIBUTED_CLUSTER_KEY] = {"cluster": cluster, "client": client}