From dae1ddd362a769d65d89bc48c457e53a4f02c1d1 Mon Sep 17 00:00:00 2001 From: Ajay Thorve Date: Tue, 13 Aug 2024 17:54:43 -0700 Subject: [PATCH] add xdist --- ci/test_external.sh | 4 ++-- ci/utils/external_dependencies.yaml | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/ci/test_external.sh b/ci/test_external.sh index 085b1b8e..244b9832 100755 --- a/ci/test_external.sh +++ b/ci/test_external.sh @@ -42,7 +42,7 @@ if [ "$PROJECT" = "all" ]; then rapids-logger "Run GPU tests for $LIBRARY" - pytest --numprocesses=8 --dist=worksteal $LIBRARY/$LIBRARY/tests --benchmark-skip --gpu + python -m pytest $LIBRARY/tests/ --numprocesses=8 --dist=worksteal --gpu popd done @@ -67,7 +67,7 @@ else rapids-logger "Run GPU tests for $LIBRARY" - pytest --numprocesses=8 --dist=worksteal $LIBRARY/$LIBRARY/tests --benchmark-skip --gpu + python -m pytest $LIBRARY/tests/ --numprocesses=8 --dist=worksteal --gpu popd fi diff --git a/ci/utils/external_dependencies.yaml b/ci/utils/external_dependencies.yaml index 518c0a96..9bc5e35e 100644 --- a/ci/utils/external_dependencies.yaml +++ b/ci/utils/external_dependencies.yaml @@ -15,3 +15,4 @@ dependencies: - dash - dask-sql - pytest-benchmark + - pytest-xdist