linting

basf · Oct 9, 2024 · 21a13bf · 21a13bf
1 parent 9c6f970
commit 21a13bf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/molpipeline/explainability/visualization/visualization.py b/molpipeline/explainability/visualization/visualization.py
@@ -453,7 +453,7 @@ def structure_heatmap_shap(
     if isinstance(explanation.prediction, float):
         # regression case
         raise NotImplementedError("Regression case not yet implemented.")
-    elif isinstance(explanation.prediction, np.ndarray):
+    if isinstance(explanation.prediction, np.ndarray):
         if len(explanation.prediction) == 2:
             # binary classification case
             text = (