Skip to content

Commit 9bfca6d

Browse files
jarverhajarverha
authored andcommitted
update test_shap_explainer to remove deeplearning
1 parent 34b7442 commit 9bfca6d

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

tests/test_shap_explainer.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
from powershap.shap_wrappers import ShapExplainerFactory
66
from powershap.shap_wrappers.shap_explainer import (
77
CatboostExplainer,
8-
DeepLearningExplainer,
8+
# DeepLearningExplainer,
99
EnsembleExplainer,
1010
LGBMExplainer,
1111
LinearExplainer,
@@ -111,11 +111,11 @@ def test_get_ensemble_explainer():
111111
assert isinstance(explainer, EnsembleExplainer)
112112

113113

114-
def test_get_deep_learning_explainer():
115-
import tensorflow as tf
114+
# def test_get_deep_learning_explainer():
115+
# import tensorflow as tf
116116

117-
explainer = ShapExplainerFactory.get_explainer(tf.keras.Sequential())
118-
assert isinstance(explainer, DeepLearningExplainer)
117+
# explainer = ShapExplainerFactory.get_explainer(tf.keras.Sequential())
118+
# assert isinstance(explainer, DeepLearningExplainer)
119119

120120

121121
def test_value_error_get_explainer():

0 commit comments

Comments
 (0)