From e075b9eaf89eeeae81611a6170eb20e4100c2703 Mon Sep 17 00:00:00 2001
From: "Wieczorek, Alexander" <Alexander.Wieczorek@empa.ch>
Date: Mon, 13 Jan 2025 20:45:40 +0100
Subject: [PATCH] Improved comments and docstring

---
 baybe/insights/shap.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/baybe/insights/shap.py b/baybe/insights/shap.py
index 79dd46e33..c30385862 100644
--- 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)