fix: Don't add unnessecary walrus parentheses

Signed-off-by: RedGuy12 <61329810+RedGuy12@users.noreply.github.com>
cobaltt7 · Jan 17, 2024 · f3bb847 · f3bb847
1 parent a8c4c9f
commit f3bb847
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/black/linegen.py b/src/black/linegen.py
@@ -242,7 +242,11 @@ def visit_dictsetmaker(self, node: Node) -> Iterator[Line]:
                 if i == 0:
                     continue
                 if node.children[i - 1].type == token.COLON:
-                    if child.type == syms.atom and child.children[0].type == token.LPAR:
+                    if (
+                        child.type == syms.atom
+                        and child.children[0].type == token.LPAR
+                        and not is_walrus_assignment(child)
+                    ):
                         if maybe_make_parens_invisible_in_atom(
                             child,
                             parent=node,
@@ -1545,7 +1549,6 @@ def maybe_make_parens_invisible_in_atom(
             and max_delimiter_priority_in_atom(node) >= COMMA_PRIORITY
         )
         or is_tuple_containing_walrus(node)
-        or is_walrus_assignment(node)
     ):
         return False
 

diff --git a/test.py b/test.py