From f8f51aa6b4af284a87526047d6c0243b35d8b2ea Mon Sep 17 00:00:00 2001 From: Martin Fitzner <17951239+Scienfitz@users.noreply.github.com> Date: Mon, 20 Jan 2025 15:04:19 +0100 Subject: [PATCH] Apply suggestions from code review Include improvements Co-authored-by: AdrianSosic --- baybe/insights/shap.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/baybe/insights/shap.py b/baybe/insights/shap.py index 63b30987a..db9d34cbe 100644 --- a/baybe/insights/shap.py +++ b/baybe/insights/shap.py @@ -360,10 +360,10 @@ def plot( plot_func = getattr(shap.plots, plot_type) # Handle plot types that only explain a single data point - if plot_type in {"force"}: + if plot_type == "force": if explanation_idx is None: warnings.warn( - f"When using plot type '{plot_type}', a 'explanation_idx' must be " + f"When using plot type '{plot_type}', an 'explanation_idx' must be " f"chosen to identify a single data point that should be explained. " f"Choosing the first measurement at position 0." )