From 1ef31c0ba5df776bc5e8e77ad0e813aaa47050f3 Mon Sep 17 00:00:00 2001
From: Jochen Sieg <jochen.sieg@basf.com>
Date: Wed, 20 Nov 2024 16:30:20 +0100
Subject: [PATCH] linting

---
 molpipeline/explainability/visualization/visualization.py       | 2 +-
 .../test_visualization/test_gaussian_grid.py                    | 1 +
 .../test_visualization/test_visualization.py                    | 1 -
 3 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/molpipeline/explainability/visualization/visualization.py b/molpipeline/explainability/visualization/visualization.py
index 3172983d..c6e06ca7 100644
--- a/molpipeline/explainability/visualization/visualization.py
+++ b/molpipeline/explainability/visualization/visualization.py
@@ -20,7 +20,7 @@
 from rdkit.Chem.Draw import rdMolDraw2D
 
 from molpipeline.abstract_pipeline_elements.core import RDKitMol
-from molpipeline.explainability import SHAPFeatureAndAtomExplanation
+from molpipeline.explainability.explanation import SHAPFeatureAndAtomExplanation
 from molpipeline.explainability.visualization.gauss import GaussFunctor2D
 from molpipeline.explainability.visualization.heatmaps import (
     ValueGrid,
diff --git a/tests/test_explainability/test_visualization/test_gaussian_grid.py b/tests/test_explainability/test_visualization/test_gaussian_grid.py
index 461186e7..40cdd20a 100644
--- a/tests/test_explainability/test_visualization/test_gaussian_grid.py
+++ b/tests/test_explainability/test_visualization/test_gaussian_grid.py
@@ -27,6 +27,7 @@
 class TestSumOfGaussiansGrid(unittest.TestCase):
     """Test sum of gaussian grid ."""
 
+    # pylint: disable=duplicate-code
     test_pipeline: ClassVar[Pipeline]
     test_explainer: ClassVar[SHAPTreeExplainer]
     test_explanations: ClassVar[
diff --git a/tests/test_explainability/test_visualization/test_visualization.py b/tests/test_explainability/test_visualization/test_visualization.py
index 1ea3896b..d9a98aba 100644
--- a/tests/test_explainability/test_visualization/test_visualization.py
+++ b/tests/test_explainability/test_visualization/test_visualization.py
@@ -5,7 +5,6 @@
 
 import numpy as np
 from rdkit import Chem
-from rdkit.Chem import Draw
 from sklearn.ensemble import RandomForestClassifier
 
 from molpipeline import Pipeline