From 21a13bf8a8e1e7bd9f6e832c979e1a0f1ecced1c Mon Sep 17 00:00:00 2001
From: Jochen Sieg <jochen.sieg@basf.com>
Date: Fri, 30 Aug 2024 17:37:39 +0200
Subject: [PATCH] linting

---
 molpipeline/explainability/visualization/visualization.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/molpipeline/explainability/visualization/visualization.py b/molpipeline/explainability/visualization/visualization.py
index 5276ec83..9a7f5d33 100644
--- 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 = (