mypy

basf · Jan 15, 2025 · ea186cc · ea186cc
1 parent 2038e29
commit ea186cc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/molpipeline/experimental/explainability/visualization/visualization.py b/molpipeline/experimental/explainability/visualization/visualization.py
@@ -288,9 +288,9 @@ def _add_shap_present_absent_features_text(
     sum_absent_shap: float
         The sum of the SHAP values for absent features.
     """
-    if explanation.prediction[-1] is None:
+    if explanation.prediction is None:
         raise AssertionError("Prediction value is None.")
-    if explanation.expected_value[-1] is None:
+    if explanation.expected_value is None:
         raise AssertionError("Expected value is None.")
 
     color1 = "black"