Fix bad error

Signed-off-by: Antoni Baum <antoni.baum@protonmail.com>
pycaret · Mar 18, 2023 · 09d2f40 · 09d2f40
1 parent bc80a3f
commit 09d2f40
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pycaret/internal/pycaret_experiment/supervised_experiment.py b/pycaret/internal/pycaret_experiment/supervised_experiment.py
@@ -4287,7 +4287,7 @@ def pdp(show: bool = True):
             try:
                 # interpret>=0.3.1
                 pdp = PartialDependence(model=model, data=test_X)
-            except AttributeError:
+            except TypeError:
                 try:
                     pdp = PartialDependence(
                         predict_fn=model.predict_proba, data=test_X