From dc87455691020581d59e305ee7120256b0ad929a Mon Sep 17 00:00:00 2001
From: jnke2016 <jnke2016@gmail.com>
Date: Tue, 3 Oct 2023 10:44:47 -0700
Subject: [PATCH] increase timeout

---
 ci/test_python.sh | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/ci/test_python.sh b/ci/test_python.sh
index 7b0077991ae..6544f969c82 100755
--- a/ci/test_python.sh
+++ b/ci/test_python.sh
@@ -65,7 +65,10 @@ popd
 
 rapids-logger "pytest cugraph"
 pushd python/cugraph/cugraph
-export DASK_WORKER_DEVICES="0"
+DASK_WORKER_DEVICES="0" \
+DASK_DISTRIBUTED__SCHEDULER__WORKER_TTL="10s" \
+DASK_DISTRIBUTED__COMM__TIMEOUTS__CONNECT="10s" \
+DASK_CUDA_WAIT_WORKERS_MIN_TIMEOUT=20 \
 pytest \
   -v \
   --benchmark-disable \