mypy

basf · Nov 15, 2024 · e3ee77d · e3ee77d
1 parent 09c8326
commit e3ee77d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_explainability/test_visualization/test_visualization.py b/tests/test_explainability/test_visualization/test_visualization.py
@@ -109,9 +109,9 @@ def test_explicit_hydrogens(self) -> None:
         self.assertEqual(len(explanations1), 1)
         self.assertEqual(len(explanations2), 1)
         self.assertEqual(len(explanations3), 1)
-        self.assertIsInstance(explanations1[0].atom_weights, np.ndarray)
-        self.assertIsInstance(explanations2[0].atom_weights, np.ndarray)
-        self.assertIsInstance(explanations3[0].atom_weights, np.ndarray)
+        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)
         self.assertEqual(len(explanations2[0].atom_weights), 1)
         self.assertEqual(len(explanations3[0].atom_weights), 1)