Fix type alias in _iter_branch

yunline · Dec 14, 2024 · 7500c77 · 7500c77
1 parent fc6c81b
commit 7500c77
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/oneliner/pending_nodes.py b/oneliner/pending_nodes.py
@@ -114,7 +114,6 @@ def get_result(self) -> list[expr]:
 
 
 class PendingExpr(PendingNode[Expr]):
-
     def get_result(self) -> list[expr]:
         return [expr_transf(self.nsp, self.node.value)]
 
@@ -136,8 +135,8 @@ def _iter_branch(
         self,
         converted_branch: list[expr],
         branch: list[stmt],
-        get_interrupt_cnt,
-        get_flow_control_expr,
+        get_interrupt_cnt: typing.Callable[[], int],
+        get_flow_control_expr: typing.Callable[[], expr],
     ) -> typing.Generator[AST, list[expr], None]:
         initial_interrupt_cnt = get_interrupt_cnt()
 
@@ -269,7 +268,6 @@ def _iter_nodes(self) -> typing.Generator[AST, list[expr], None]:
 
 
 class PendingWhile(_PendingLoop[While]):
-
     def __init__(self, node: While, nsp: Namespace, nsp_global: NamespaceGlobal):
         super().__init__(node, nsp, nsp_global)
 
@@ -392,7 +390,6 @@ def get_result(self) -> list[expr]:
 
 
 class PendingFor(_PendingLoop[For]):
-
     def __init__(self, node: For, nsp: Namespace, nsp_global: NamespaceGlobal):
         super().__init__(node, nsp, nsp_global)