From f0b6aa3cea34fd1af425ba412cfcc6aa883863f4 Mon Sep 17 00:00:00 2001
From: Bradley Dice <bdice@bradleydice.com>
Date: Tue, 5 Mar 2024 08:03:57 -0600
Subject: [PATCH] Switch `pytest-xdist` algorithm to `worksteal`

---
 ci/test_external.sh | 1 +
 ci/test_python.sh   | 1 +
 2 files changed, 2 insertions(+)

diff --git a/ci/test_external.sh b/ci/test_external.sh
index 87b68dc3..9d1ada44 100755
--- a/ci/test_external.sh
+++ b/ci/test_external.sh
@@ -81,6 +81,7 @@ set +e
 
 rapids-logger "running all gathered tests"
 DATASHADER_TEST_GPU=1 pytest --numprocesses=8 $FILES
+ --dist=worksteal \
 
 if [[ "$PROJECT" = "all" ]] || [[ "$PROJECT" = "datashader" ]]; then
     # run test_quadmesh.py separately as dask.array tests fail with numprocesses
diff --git a/ci/test_python.sh b/ci/test_python.sh
index af505feb..50bc0817 100755
--- a/ci/test_python.sh
+++ b/ci/test_python.sh
@@ -44,6 +44,7 @@ pytest \
   --cache-clear \
   --junitxml="${RAPIDS_TESTS_DIR}/junit-cuxfilter.xml" \
    --numprocesses=8 \
+   --dist=worksteal \
   --cov-config=.coveragerc \
   --cov=cuxfilter \
   --cov-report=xml:"${RAPIDS_COVERAGE_DIR}/cuxfilter-coverage.xml" \