Parse complex expressions in parameters after * and **

[etc/vim.git] / black.py

diff --git a/black.py b/black.py
index c3610c74b12a43ecf654a9e08cadaa3cc8f9a5f7..7c487f6e938c1b8113674b10bc91877da0546aa3 100644 (file)
--- a/black.py
+++ b/black.py
@@ -1364,7 +1364,7 @@ def whitespace(leaf: Leaf) -> str:  # noqa C901
             if not prevp or prevp.type == token.LPAR:
                 return NO
 
             if not prevp or prevp.type == token.LPAR:
                 return NO
 

-        elif prev.type == token.EQUAL or prev.type == token.DOUBLESTAR:
+        elif prev.type in {token.EQUAL, token.STAR, token.DOUBLESTAR}:

             return NO
 
     elif p.type == syms.decorator:
             return NO
 
     elif p.type == syms.decorator: