Skip to content

Commit 449173c

Browse files
authored
Run cudf-polars wheels unit tests with more than 1 process (#20124)
Follow up to #19980 Authors: - Matthew Roeschke (https://github.com/mroeschke) Approvers: - Bradley Dice (https://github.com/bdice) - Vyas Ramasubramani (https://github.com/vyasr) URL: #20124
1 parent d3e98c9 commit 449173c

File tree

3 files changed

+5
-1
lines changed

3 files changed

+5
-1
lines changed

ci/test_python_other.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ rapids-logger "pytest custreamz"
4343
rapids-logger "pytest cudf-polars"
4444
./ci/run_cudf_polars_pytests.sh \
4545
--junitxml="${RAPIDS_TESTS_DIR}/junit-cudf-polars.xml" \
46-
--numprocesses=6 \
46+
--numprocesses=8 \
4747
--dist=worksteal \
4848
--cov-config=./pyproject.toml \
4949
--cov=cudf_polars \

ci/test_wheel_cudf_polars.sh

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,8 @@ for version in "${VERSIONS[@]}"; do
7171

7272
./ci/run_cudf_polars_pytests.sh \
7373
"${COVERAGE_ARGS[@]}" \
74+
--numprocesses=8 \
75+
--dist=worksteal \
7476
--junitxml="${RAPIDS_TESTS_DIR}/junit-cudf-polars-${version}.xml"
7577

7678
if [ $? -ne 0 ]; then

python/cudf_polars/tests/experimental/conftest.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,8 @@
77
import pytest
88

99

10+
# scope="session" is important to not cause singificant slowdowns in CI
11+
# https://github.com/rapidsai/cudf/pull/20137
1012
@pytest.fixture(autouse=True, scope="session")
1113
def dask_cluster(pytestconfig, worker_id):
1214
if (

0 commit comments

Comments
 (0)