Lint and import fixes

Author: vivekmig

captum/insights/attr_vis/features.py

@@ -8,7 +8,7 @@
 from captum._utils.common import safe_div
 from captum.attr._utils import visualization as viz
 from captum.insights.attr_vis._utils.transforms import format_transforms
-from torch._tensor import Tensor
+from torch import Tensor
 
 FeatureOutput = namedtuple("FeatureOutput", "name base modified type contribution")
 

setup.py

@@ -67,7 +67,7 @@ def report(*args):
     INSIGHTS_REQUIRES
     + TEST_REQUIRES
     + [
-        "black==22.3.0",
+        "black==24.2.0",
         "flake8",
         "sphinx",
         "sphinx-autodoc-typehints",

tests/attr/helpers/conductance_reference.py

@@ -10,7 +10,7 @@
 from captum.attr._utils.approximation_methods import approximation_parameters
 from captum.attr._utils.attribution import LayerAttribution
 from captum.attr._utils.common import _reshape_and_sum
-from torch._tensor import Tensor
+from torch import Tensor
 
 """
 Note: This implementation of conductance follows the procedure described in the original

tests/attr/layer/test_layer_lrp.py

@@ -9,7 +9,7 @@
 
 from tests.helpers.basic import assertTensorAlmostEqual, BaseTest
 from tests.helpers.basic_models import BasicModel_ConvNet_One_Conv, SimpleLRPModel
-from torch._tensor import Tensor
+from torch import Tensor
 
 
 def _get_basic_config() -> Tuple[BasicModel_ConvNet_One_Conv, Tensor]:

tests/attr/test_guided_grad_cam.py

@@ -8,7 +8,7 @@
 from captum.attr._core.guided_grad_cam import GuidedGradCam
 from tests.helpers.basic import assertTensorAlmostEqual, BaseTest
 from tests.helpers.basic_models import BasicModel_ConvNet_One_Conv
-from torch._tensor import Tensor
+from torch import Tensor
 from torch.nn import Module
 
 

tests/attr/test_interpretable_input.py

@@ -4,7 +4,7 @@
 from captum.attr._utils.interpretable_input import TextTemplateInput, TextTokenInput
 from parameterized import parameterized
 from tests.helpers.basic import assertTensorAlmostEqual, BaseTest
-from torch._tensor import Tensor
+from torch import Tensor
 
 
 class DummyTokenizer:

tests/utils/test_linear_model.py

@@ -9,7 +9,7 @@
     SGDRidge,
 )
 from tests.helpers.basic import assertTensorAlmostEqual, BaseTest
-from torch._tensor import Tensor
+from torch import Tensor
 
 
 def _evaluate(test_data, classifier) -> Dict[str, float]: