From e880e6fbdfbf5078dc96092a24f59bb34801a1b6 Mon Sep 17 00:00:00 2001
From: j-ittner <ittner.jan@bcg.com>
Date: Tue, 11 Jul 2023 00:08:59 +0200
Subject: [PATCH] TEST: only re-fit the model where needed

---
 test/test/facet/test_inspection.py | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/test/test/facet/test_inspection.py b/test/test/facet/test_inspection.py
index 5c2985f4..f59a8eb6 100644
--- a/test/test/facet/test_inspection.py
+++ b/test/test/facet/test_inspection.py
@@ -126,7 +126,7 @@ def test_model_inspection(
             assert (
                 sample.features.notna().all().all()
             ), "observations must not contain missing values"
-            model = regressor.fit(X=sample.features, y=sample.target)
+            model = regressor
             regressor_feature_names = set(sample.feature_names)
 
         else:
@@ -146,10 +146,7 @@ def test_model_inspection(
 
         # noinspection PyTypeChecker
         inspector = NativeLearnerInspector(
-            model=(
-                # fit the model on the sample
-                model.fit(X=sample.features, y=sample.target)
-            ),
+            model=model,
             explainer_factory=explainer_factory,
             n_jobs=n_jobs,
         ).fit(sample)