18 buildPythonPackage rec {
23 src = fetchFromGitHub {
27 sha256 = "1wxnxvbz6avzzfqjfbcqd4v879hvpq4021v31fhdpccr2q317rr9";
30 propagatedBuildInputs = [
42 # when importing the local copy the extension is not found
45 checkInputs = [ pytestCheckHook matplotlib nose ipython ];
46 # Those tests access the network
48 "test_kernel_shap_with_a1a_sparse_zero_background"
49 "test_kernel_shap_with_a1a_sparse_nonzero_background"
50 "test_kernel_shap_with_high_dim_sparse"
51 "test_sklearn_random_forest_newsgroups"
52 "test_sum_match_random_forest"
53 "test_sum_match_extra_trees"
54 "test_single_row_random_forest"
55 "test_sum_match_gradient_boosting_classifier"
56 "test_single_row_gradient_boosting_classifier"
57 "test_HistGradientBoostingClassifier_proba"
58 "test_HistGradientBoostingClassifier_multidim"
59 "test_sum_match_gradient_boosting_regressor"
60 "test_single_row_gradient_boosting_regressor"
64 description = "A unified approach to explain the output of any machine learning model";
65 homepage = "https://github.com/slundberg/shap";
66 license = licenses.mit;
67 maintainers = with maintainers; [ evax ];
68 platforms = platforms.unix;
69 # ModuleNotFoundError: No module named 'sklearn.ensemble.iforest'