Ignore non-aliased module nodes when collecting dependencies.

rchen152 · rchen152 · commit 6bfe898d7ec4 · 2024-02-09T16:34:41.000-08:00
This fixes a failure in typeshed's pytype_test.

PiperOrigin-RevId: 605760245
diff --git a/pytype/pytd/visitors.py b/pytype/pytd/visitors.py
@@ -1536,7 +1536,12 @@ def EnterLateType(self, node):
     self._ProcessName(node.name, self.late_dependencies)
 
   def EnterModule(self, node):
-    self._ProcessName(node.module_name, self.dependencies)
+    # Most module nodes look like:
+    # Module(name='foo_module.bar_module', module_name='bar_module').
+    # We don't care about these. Nodes that don't follow this pattern are
+    # aliased modules, which we need to record.
+    if not node.name.endswith("." + node.module_name):
+      self._ProcessName(node.module_name, self.dependencies)
 
 
 def ExpandSignature(sig):
