Switch pytest-xdist algorithm to worksteal

rapidsai · Mar 5, 2024 · f0b6aa3 · f0b6aa3
1 parent e45ba7e
commit f0b6aa3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git 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
@@ -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" \