Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
28 commits
Select commit Hold shift + click to select a range
3d547d0
Add optional deps for sklearn examples
csadorf May 12, 2026
b17822a
Enforce zero sklearn example xfails
csadorf May 12, 2026
d31cf15
Remove sklearn example xfail ratchets
csadorf Jun 12, 2026
7463e87
Narrow PR workflow for sklearn debug
csadorf Jun 12, 2026
551a1f7
Fallback PCA unsupported configurations to CPU
csadorf Jun 12, 2026
cfb104e
Fallback KernelRidge sklearn kernels to CPU
csadorf Jun 12, 2026
150d548
Keep ColumnTransformer pipeline data on host
csadorf Jun 12, 2026
3bde04b
revert this after debugging: increase sklearn example timeouts
csadorf Jun 12, 2026
425514c
Xfail upstream wikipedia sklearn example
csadorf Jun 12, 2026
7d73e12
Update PCA sklearn accel xfails
csadorf Jun 22, 2026
25d12fc
Revert cuml implementation changes
csadorf Jun 26, 2026
7b8ef6e
Install sklearn example extras in CI script
csadorf Jun 26, 2026
64f1ff6
Drop moved sklearn example test changes
csadorf Jun 26, 2026
e5cca19
Fix PR workflow shared-workflows refs
csadorf Jun 26, 2026
51d16bb
Log sklearn example timings in CI
csadorf Jun 26, 2026
e627ab2
Update PCA randomized sklearn xfails
csadorf Jun 26, 2026
4e13cc6
Preserve network failures as xfails
csadorf Jul 6, 2026
e8a0a67
Restore sklearn upstream xfail list
csadorf Jul 6, 2026
8dca52c
Document sklearn example runner usage
csadorf Jul 7, 2026
df13b27
Update upstream accel test README
csadorf Jul 7, 2026
5c86288
Preserve sklearn example timeouts as xfails
csadorf Jul 7, 2026
3044214
Raise sklearn examples pass-rate gate
csadorf Jul 7, 2026
9d73463
Fix upstream summary pass-rate accounting
csadorf Jul 7, 2026
b144257
Reduce sklearn example timeout cap
csadorf Jul 7, 2026
4391f52
Lower sklearn examples pass-rate gate
csadorf Jul 7, 2026
4835366
Restore PR workflow configuration
csadorf Jul 7, 2026
4eed7aa
Merge remote-tracking branch 'origin/main' into pr-8090
csadorf Jul 7, 2026
74b3d9a
Simplify sklearn examples CI exit handling
csadorf Jul 7, 2026
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
19 changes: 14 additions & 5 deletions ci/test_python_sklearn_examples.sh
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,13 @@ cd "$(dirname "$(realpath "${BASH_SOURCE[0]}")")"/../ || exit 1
export DEPENDENCY_FILE_KEY=test_python_accel_sklearn
source ./ci/test_python_common.sh

rapids-logger "Install optional scikit-learn example dependencies"
rapids-mamba-retry install --yes -n test \
"plotly>=6,<7" \
"polars>=1,<2" \
"pooch>=1,<2" \
scikit-image

EXITCODE=0
trap "EXITCODE=1" ERR
set +e
Expand All @@ -18,15 +25,17 @@ SKLEARN_EXAMPLES_JUNITXML="${RAPIDS_TESTS_DIR}/junit-sklearn-examples.xml"
# Run scikit-learn examples under cuml.accel
rapids-logger "scikit-learn examples"
timeout -v --signal=SIGINT --kill-after=60s 60m ./python/cuml/cuml_accel_tests/upstream/scikit-learn/run-examples.sh \
-vv --durations=0 --durations-min=0 \
-n 4 --dist worksteal \
--example-timeout=300 \
--junitxml="${SKLEARN_EXAMPLES_JUNITXML}"

# The examples tests tolerate timeouts and network issues (warn-only), but
# require a healthy majority of examples to pass so widespread regressions are
# not missed.
rapids-logger "scikit-learn examples: require >=50% pass rate"
# Per-example timeouts and network failures are reported as xfails. The
# examples tests still require a healthy majority of examples to pass so
# widespread regressions are not missed.
rapids-logger "scikit-learn examples: require >=90% pass rate"
./python/cuml/cuml_accel_tests/upstream/summarize-results.py \
--fail-below 50 \
--fail-below 90 \
"${SKLEARN_EXAMPLES_JUNITXML}"

rapids-logger "Test script exiting with value: $EXITCODE"
Expand Down
Loading
Loading