Skip to content

Commit 41b44dd

Browse files
jarverhajarverha
authored andcommitted
fixing some pipeline tests
1 parent aa962e6 commit 41b44dd

File tree

2 files changed

+4
-6
lines changed

2 files changed

+4
-6
lines changed

tests/test_pipeline_powershap.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -177,7 +177,7 @@ def test_pipeline_catboost_class_standardscaler_powershap(dummy_classification):
177177
assert n_informative > 0, "No informative columns in the dummy data!"
178178

179179
selector = PowerShap(
180-
model=make_pipeline(StandardScaler, CatBoostClassifier(n_estimators=250, verbose=0)), power_iterations=15, automatic=False
180+
model=make_pipeline(StandardScaler(), CatBoostClassifier(n_estimators=250, verbose=0)), power_iterations=15, automatic=False
181181
)
182182

183183
selector.fit(X, y)
@@ -194,7 +194,7 @@ def test_pipeline_catboost_class_maxabsscalerr_robustscaler_powershap(dummy_clas
194194
assert n_informative > 0, "No informative columns in the dummy data!"
195195

196196
selector = PowerShap(
197-
model=make_pipeline(MaxAbsScaler, RobustScaler, CatBoostClassifier(n_estimators=250, verbose=0)), power_iterations=15, automatic=False
197+
model=make_pipeline(MaxAbsScaler(), RobustScaler(), CatBoostClassifier(n_estimators=250, verbose=0)), power_iterations=15, automatic=False
198198
)
199199

200200
selector.fit(X, y)

tests/test_shap_explainer.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -167,10 +167,8 @@ def test_get_pipeline_explainer():
167167

168168
for model_class in model_classes:
169169
DummyScaler = FunctionTransformer(lambda x: x)
170-
171-
make_pipeline(DummyScaler, model_class)
172-
173-
explainer = ShapExplainerFactory.get_explainer(make_pipeline)
170+
explainer = ShapExplainerFactory.get_explainer(make_pipeline(DummyScaler, model_class))
171+
174172
assert isinstance(explainer, PipelineExplainer)
175173

176174

0 commit comments

Comments
 (0)