types

psf · Jan 28, 2022 · ae71c47 · ae71c47
1 parent 16cbced
commit ae71c47
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/black/linegen.py b/src/black/linegen.py
@@ -156,7 +156,7 @@ def visit_suite(self, node: Node) -> Iterator[Line]:
 
     def visit_simple_stmt(self, node: Node) -> Iterator[Line]:
         """Visit a statement without nested statements."""
-        prev_type = None
+        prev_type: Optional[int] = None
         for child in node.children:
             if (prev_type is None or prev_type == token.SEMI) and is_arith_like(child):
                 wrap_in_parentheses(node, child, visible=False)

diff --git a/src/black/nodes.py b/src/black/nodes.py
@@ -536,7 +536,7 @@ def first_leaf_column(node: Node) -> Optional[int]:
     return None
 
 
-def is_arith_like(node: Node) -> bool:
+def is_arith_like(node: LN) -> bool:
     """Whether node is an arithmetic or a binary arithmetic expression"""
     return node.type in {
         syms.arith_expr,