Fix plots

Signed-off-by: Antoni Baum <antoni.baum@protonmail.com>
pycaret · Mar 18, 2023 · cfc1ef3 · cfc1ef3
1 parent 8e8e7ce
commit cfc1ef3
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/pycaret/internal/pycaret_experiment/supervised_experiment.py b/pycaret/internal/pycaret_experiment/supervised_experiment.py
@@ -4286,7 +4286,11 @@ def pdp(show: bool = True):
 
             try:
                 # interpret>=0.3.1
-                pdp = PartialDependence(model=model, data=test_X)
+                pdp = PartialDependence(
+                    model=model,
+                    data=test_X.to_numpy(),
+                    feature_names=list(test_X.columns),
+                )
             except TypeError:
                 try:
                     pdp = PartialDependence(
@@ -4314,7 +4318,11 @@ def msa(show: bool = True):
 
             try:
                 # interpret>=0.3.1
-                msa = MorrisSensitivity(model=model, data=test_X)
+                msa = MorrisSensitivity(
+                    model=model,
+                    data=test_X.to_numpy(),
+                    feature_names=list(test_X.columns),
+                )
             except TypeError:
                 try:
                     msa = MorrisSensitivity(