fix mypy

vyperlang · Feb 3, 2024 · 82e1da0 · 82e1da0
1 parent d82eb3e
commit 82e1da0
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 3 deletions.
diff --git a/vyper/ast/nodes.pyi b/vyper/ast/nodes.pyi
@@ -248,6 +248,12 @@ class ImplementsDecl(VyperNode):
     target: Name = ...
     annotation: Name = ...
 
+class UsesDecl(VyperNode):
+    annotation: VyperNode = ...
+
+class InitializesDecl(VyperNode):
+    annotation: VyperNode = ...
+
 class If(VyperNode):
     body: list = ...
     orelse: list = ...

diff --git a/vyper/codegen/function_definitions/__init__.py b/vyper/codegen/function_definitions/__init__.py
@@ -1,4 +1,4 @@
 from .external_function import generate_ir_for_external_function
 from .internal_function import generate_ir_for_internal_function
 
-__all__ = [generate_ir_for_internal_function, generate_ir_for_external_function]
+__all__ = [generate_ir_for_internal_function, generate_ir_for_external_function]  # type: ignore
diff --git a/vyper/codegen/function_definitions/external_function.py b/vyper/codegen/function_definitions/external_function.py
@@ -58,7 +58,7 @@ def _register_function_args(func_t: ContractFunctionT, context: Context) -> list
 
 def _generate_kwarg_handlers(
     func_t: ContractFunctionT, context: Context
-) -> dict[str, tuple[int, IRnode]]:
+) -> dict[str, EntryPointInfo]:
     # generate kwarg handlers.
     # since they might come in thru calldata or be default,
     # allocate them in memory and then fill it in based on calldata or default,

diff --git a/vyper/codegen/function_definitions/internal_function.py b/vyper/codegen/function_definitions/internal_function.py
@@ -11,7 +11,7 @@
 
 def generate_ir_for_internal_function(
     code: vy_ast.FunctionDef, module_ctx, is_ctor_context: bool
-) -> IRnode:
+) -> InternalFuncIR:
     """
     Parse a internal function (FuncDef), and produce full function body.