Merge pull request #1237 from rapidsai/branch-23.10

Forward-merge branch-23.10 to branch-23.12
rapidsai · Sep 25, 2023 · a053157 · a053157
2 parents 66c1907 + ec80f97
commit a053157
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/dask_cuda/tests/test_proxy.py b/dask_cuda/tests/test_proxy.py
@@ -400,7 +400,7 @@ def _pxy_deserialize(self):
 
 @pytest.mark.parametrize("send_serializers", [None, ("dask", "pickle"), ("cuda",)])
 @pytest.mark.parametrize("protocol", ["tcp", "ucx"])
-@gen_test(timeout=60)
+@gen_test(timeout=120)
 async def test_communicating_proxy_objects(protocol, send_serializers):
     """Testing serialization of cuDF dataframe when communicating"""
     cudf = pytest.importorskip("cudf")

diff --git a/dask_cuda/tests/test_spill.py b/dask_cuda/tests/test_spill.py
@@ -121,7 +121,7 @@ def delayed_worker_assert(
         },
     ],
 )
-@gen_test(timeout=30)
+@gen_test(timeout=120)
 async def test_cupy_cluster_device_spill(params):
     cupy = pytest.importorskip("cupy")
     with dask.config.set({"distributed.worker.memory.terminate": False}):
@@ -212,7 +212,7 @@ async def test_cupy_cluster_device_spill(params):
         },
     ],
 )
-@gen_test(timeout=30)
+@gen_test(timeout=120)
 async def test_cudf_cluster_device_spill(params):
     cudf = pytest.importorskip("cudf")