Files

T

History

Tobias Pitters c3481c11aa Refactor short circuiting of tree_path_dependent shap calculation (#5069 )

* refactor internal short circuiting for internal tree model shap calculation

* refactor

* remove unnecessary if statement

* add version check for categorical xgboost handling

* regenerate images

* update selenium-generated images

* limit numpy so that more recent version of numba is installed

* fix mypy tests, unpin gpboost

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* ignore mypy error for cutils

* add actions to all

* Make Selenium force-plot baseline comparison tolerant to Chrome version drift

Chrome/Chromedriver versions differ between local machines and CI runners
(both unpinned), which can shift the rendered screenshot size by a few
pixels with no actual change to the plot, causing spurious failures. Allow
a small size tolerance and diff over the shared cropped region instead of
requiring an exact dimension match.

Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>

* revert to a working state for maskers, ignore the line in mypy

* fix numba issue reference in pyproject.toml

* add graphviz to fix failing notebooks

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Claude Sonnet 5 <noreply@anthropic.com>

2026-07-20 14:59:00 +02:00

actions

Merge branch 'master' into fix/actions-test-coverage

2026-04-16 01:00:27 +05:30

benchmark

Maint: apply ruff formatter to all python files (#3722 )

2024-06-28 17:12:55 +01:00

explainers

Refactor short circuiting of tree_path_dependent shap calculation (#5069 )

2026-07-20 14:59:00 +02:00

maskers

TST: Increase test coverage for shap/maskers/_image.py (#4519 )

2026-06-04 22:16:59 +02:00

models

Move to nanobind (#4366 )

2026-04-17 22:35:19 +02:00

plots

Refactor short circuiting of tree_path_dependent shap calculation (#5069 )

2026-07-20 14:59:00 +02:00

shap_values_baselines

Move to nanobind (#4366 )

2026-04-17 22:35:19 +02:00

utils

Fix IndexError in format_value() when handling empty strings (#4238 )

2025-12-11 21:18:51 +01:00

conftest.py

Move to nanobind (#4366 )

2026-04-17 22:35:19 +02:00

datasets_to_cache.py

fix macos tests (#4178 )

2025-10-14 08:52:59 +02:00

gpu_tree_tests.ipynb

run gpu_tree_tests.ipynb in order to fix pre-commit error (#4277 )

2026-02-19 21:03:12 +01:00

README.md

Add Colab Notebook to test GPUTreeExplainer (#4266 )

2026-02-11 22:57:23 +01:00

test_datasets.py

DOCS: Enable subset of pydocstyle rules, bulk update ruff fixes (#3550 )

2024-03-08 11:08:29 +00:00

test_explanation.py

fix: record "sample" in op_history instead of "__getitem__" (#4376 )

2026-04-21 18:16:48 +02:00

test_install.py

Maint: apply ruff formatter to all python files (#3722 )

2024-06-28 17:12:55 +01:00

README.md

SHAP Tests

GPU TreeExplainer Tests

The GPU TreeExplainer tests require a CUDA-capable GPU and the _cext_gpu extension built from source. Since GPU hardware is not available in our CI pipeline, these tests can be run manually on Google Colab.

Running on Google Colab

Click the badge above to open the notebook in Google Colab
Switch to a GPU runtime: Runtime > Change runtime type > T4 GPU
Run all cells