py39 compat

PyCQA · Jul 9, 2023 · 17698fa · 17698fa
1 parent 20af399
commit 17698fa
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pyi.py b/pyi.py
@@ -971,6 +971,11 @@ def visit_ImportFrom(self, node: ast.ImportFrom) -> None:
         if module_name is None:
             return
 
+        if sys.version_info < (3, 10):
+            for obj in node.names:
+                obj.lineno = node.lineno
+                obj.col_offset = node.col_offset
+
         imported_names = {obj.name: obj for obj in node.names}
 
         if module_name == "__future__":