Fix lint

Author: vivekmig

tests/attr/helpers/conductance_reference.py

@@ -1,5 +1,5 @@
 #!/usr/bin/env python3
-from typing import cast, Optional, Tuple, Union
+from typing import cast, Tuple, Union
 
 import numpy as np
 import torch

tests/attr/models/test_pytext.py

@@ -5,7 +5,7 @@
 import os
 import tempfile
 import unittest
-from typing import Dict, List, NoReturn, Optional
+from typing import Dict, List
 
 import torch
 

tests/helpers/basic.py

@@ -2,12 +2,11 @@
 import copy
 import random
 import unittest
-from typing import Callable, List, Tuple, Union
+from typing import Callable
 
 import numpy as np
 import torch
 from captum.log import patch_methods
-from torch import Tensor
 
 
 def deep_copy_args(func: Callable):

tests/robust/test_FGSM.py

@@ -2,7 +2,7 @@
 from typing import Any, Callable, List, Optional, Tuple, Union
 
 import torch
-from captum._utils.typing import TensorLikeList, TensorOrTupleOfTensorsGeneric
+from captum._utils.typing import TensorOrTupleOfTensorsGeneric
 from captum.robust import FGSM
 from tests.helpers.basic import assertTensorAlmostEqual, BaseTest
 from tests.helpers.basic_models import BasicModel, BasicModel2, BasicModel_MultiLayer

tests/utils/test_sample_gradient.py

@@ -139,7 +139,9 @@ def test_sample_grads_layer_modules(self) -> None:
                     # So, check that we did calculate sample grads for the desired
                     # layers via the above checking approach.
                     for parameter in module.parameters():
-                        assert not isinstance(parameter.sample_grad, int)  # type: ignore
+                        assert not isinstance(
+                            parameter.sample_grad, int  # type: ignore
+                        )
                 else:
                     # For the layers we do not want sample grads for, their
                     # `sample_grad` should still be 0, since they should not have been