Improve docstrings

emdgroup · Jan 14, 2025 · cc41592 · cc41592
1 parent cb86db4
commit cc41592
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/baybe/insights/shap.py b/baybe/insights/shap.py
@@ -114,7 +114,7 @@ def model(x: np.ndarray) -> np.ndarray:
 class SHAPInsight:
     """Class for SHAP-based feature importance insights.
 
-    Also supports LIME and MAPLE explainers via the ``shap`` module.
+    Also supports LIME and MAPLE explainers via the ``shap`` package.
     """
 
     explainer: shap.Explainer = field(validator=instance_of(shap.Explainer))
@@ -172,7 +172,7 @@ def from_campaign(
         Uses the measurements of the campaign as background data.
 
         Args:
-            campaign: A campaign holding a recommender using a surrogate model.
+            campaign: A campaign using a surrogate-based recommender.
             explainer_cls: See :func:`make_explainer_for_surrogate`.
             use_comp_rep: See :func:`make_explainer_for_surrogate`.