Improved comments and docstring

emdgroup · Jan 13, 2025 · e075b9e · e075b9e
1 parent 03c95f7
commit e075b9e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/baybe/insights/shap.py b/baybe/insights/shap.py
@@ -145,7 +145,7 @@ def from_surrogate(
         explainer_cls: type[shap.Explainer] | str = _DEFAULT_EXPLAINER_CLS,
         use_comp_rep: bool = False,
     ):
-        """Create a SHAP insight from a campaign.
+        """Create a SHAP insight from a surrogate.
 
         For details, see :func:`make_explainer_for_surrogate`.
         """
@@ -320,6 +320,7 @@ def plot(
         if data is None:
             data = self.background_data
 
+        # Use custom scatter plot function to ignore non-numeric features
         if plot_type == "scatter":
             return self._plot_shap_scatter(data, show=show, **kwargs)