linting

basf · Nov 20, 2024 · 1ef31c0 · 1ef31c0
1 parent a0ed00e
commit 1ef31c0
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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
@@ -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