chore: loosen dependency on asttokens (#3149)

vyperlang · Nov 14, 2022 · 46b6cfe · 46b6cfe
1 parent 886488f
commit 46b6cfe
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -92,7 +92,7 @@ def _global_version(version):
     python_requires=">=3.7,<3.11",
     py_modules=["vyper"],
     install_requires=[
-        "asttokens==2.0.5",
+        "asttokens>=2.0.5,<3",
         "pycryptodome>=3.5.1,<4",
         "semantic-version>=2.10,<3",
         "cached-property==1.5.2 ; python_version<'3.8'",

diff --git a/vyper/ast/annotation.py b/vyper/ast/annotation.py
@@ -1,7 +1,7 @@
 import ast as python_ast
 import tokenize
 from decimal import Decimal
-from typing import Optional
+from typing import Optional, cast
 
 import asttokens
 
@@ -274,7 +274,7 @@ def annotate_python_ast(
         The annotated and optimized AST.
     """
 
-    tokens = asttokens.ASTTokens(source_code, tree=parsed_ast)
+    tokens = asttokens.ASTTokens(source_code, tree=cast(Optional[python_ast.Module], parsed_ast))
     visitor = AnnotatingVisitor(source_code, modification_offsets, tokens, source_id, contract_name)
     visitor.visit(parsed_ast)