linting

basf · Nov 25, 2024 · f12432a · f12432a
1 parent 725cdb8
commit f12432a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/test_explainability/test_visualization/test_visualization.py b/tests/test_explainability/test_visualization/test_visualization.py
@@ -149,17 +149,17 @@ def test_explicit_hydrogens(self) -> None:
         self.assertIsInstance(explanations1[0].atom_weights, np.ndarray)  # type: ignore[union-attr]
         self.assertIsInstance(explanations2[0].atom_weights, np.ndarray)  # type: ignore[union-attr]
         self.assertIsInstance(explanations3[0].atom_weights, np.ndarray)  # type: ignore[union-attr]
-        self.assertEqual(len(explanations1[0].atom_weights), 1)  # type: ignore[union-attr]
-        self.assertEqual(len(explanations2[0].atom_weights), 1)  # type: ignore[union-attr]
-        self.assertEqual(len(explanations3[0].atom_weights), 1)  # type: ignore[union-attr]
+        self.assertEqual(len(explanations1[0].atom_weights), 1)  # type: ignore
+        self.assertEqual(len(explanations2[0].atom_weights), 1)  # type: ignore
+        self.assertEqual(len(explanations3[0].atom_weights), 1)  # type: ignore
 
         # test visualization
         all_explanations = explanations1 + explanations2 + explanations3
         for explanation in all_explanations:
             self.assertTrue(explanation.is_valid())
             image = structure_heatmap(
                 explanation.molecule,
-                explanation.atom_weights,  # type: ignore[union-attr, arg-type]
+                explanation.atom_weights,  # type: ignore
                 width=128,
                 height=128,
             )  # type: ignore[union-attr]