diff --git a/dask_cuda/tests/test_proxy.py b/dask_cuda/tests/test_proxy.py index 53282bef1..c779a39ef 100644 --- 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 index cd36cb781..859f55d99 100644 --- 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")