increase timeout

rapidsai · Oct 3, 2023 · dc87455 · dc87455
1 parent c00c3f1
commit dc87455
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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 \