From f12432aab0ed57ac01ae59a7c69d9afa53720939 Mon Sep 17 00:00:00 2001 From: Jochen Sieg Date: Mon, 25 Nov 2024 15:13:12 +0100 Subject: [PATCH] linting --- .../test_visualization/test_visualization.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/test_explainability/test_visualization/test_visualization.py b/tests/test_explainability/test_visualization/test_visualization.py index 40db2822..21b0c57c 100644 --- a/tests/test_explainability/test_visualization/test_visualization.py +++ b/tests/test_explainability/test_visualization/test_visualization.py @@ -149,9 +149,9 @@ 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 @@ -159,7 +159,7 @@ def test_explicit_hydrogens(self) -> None: 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]