[TorchFix] Fix bug in TorchVisionDeprecatedToTensorVisitor with ImportStar (#4641)

Incrementing patch version as 0.1.0 already pushed to PyPI.

Author: kit1980

tests/fixtures/vision/checker/to_tensor.py

@@ -24,3 +24,5 @@
 import torchvision.transforms.v2
 
 torchvision.transforms.v2.ToTensor
+
+from torchvision.transforms.v2 import *

torchfix/torchfix.py

@@ -13,10 +13,11 @@
 from .visitors.performance import TorchSynchronizedDataLoaderVisitor
 from .visitors.misc import TorchRequireGradVisitor
 from .visitors.vision import (
-    TorchVisionDeprecatedPretrainedVisitor, TorchVisionDeprecatedToTensorVisitor
+    TorchVisionDeprecatedPretrainedVisitor,
+    TorchVisionDeprecatedToTensorVisitor,
 )
 
-__version__ = "0.1.0"
+__version__ = "0.1.1"
 
 DEPRECATED_CONFIG_PATH = Path(__file__).absolute().parent / "deprecated_symbols.yaml"
 

torchfix/visitors/vision/to_tensor.py

@@ -1,3 +1,4 @@
+from collections.abc import Sequence
 import libcst as cst
 
 from ...common import LintViolation, TorchVisitor
@@ -34,10 +35,11 @@ def visit_ImportFrom(self, node):
         if module_path is None:
             return
 
-        for import_node in node.names:
-            self._maybe_add_violation(
-                f"{module_path}.{import_node.evaluated_name}", import_node
-            )
+        if isinstance(node.names, Sequence):
+            for import_node in node.names:
+                self._maybe_add_violation(
+                    f"{module_path}.{import_node.evaluated_name}", import_node
+                )
 
     def visit_Attribute(self, node):
         qualified_names = self.get_metadata(cst.metadata.QualifiedNameProvider, node)