From ec80f97b2f01328df6512c9cc1379a784e13f0b9 Mon Sep 17 00:00:00 2001 From: Peter Andreas Entschev Date: Mon, 25 Sep 2023 23:08:36 +0200 Subject: [PATCH] Increase test timeouts further to reduce CI failures (#1234) Authors: - Peter Andreas Entschev (https://github.com/pentschev) Approvers: - GALI PREM SAGAR (https://github.com/galipremsagar) URL: https://github.com/rapidsai/dask-cuda/pull/1234 --- dask_cuda/tests/test_proxy.py | 2 +- dask_cuda/tests/test_spill.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) 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")