Filter data to measurement parameters only when initializing from rec…

…ommender
emdgroup · Jan 13, 2025 · bf9cf2b · bf9cf2b
1 parent e075b9e
commit bf9cf2b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/baybe/insights/shap.py b/baybe/insights/shap.py
@@ -224,9 +224,11 @@ def from_recommender(
             searchspace, objective, measurements
         )
 
+        data = measurements[[p.name for p in searchspace.parameters]]
+
         return cls.from_surrogate(
             surrogate_model,
-            searchspace.transform(measurements) if use_comp_rep else measurements,
+            searchspace.transform(data) if use_comp_rep else data,
             explainer_cls=explainer_cls,
             use_comp_rep=use_comp_rep,
         )