Fix crash with walrus + await + with (#3473)

[etc/vim.git] / src / black / lines.py

diff --git a/src/black/lines.py b/src/black/lines.py
index 08281bcf370a1a987fcffa9602d54a7fe3d71593..2aa675c3b31c21b8c38438916387da56807499c6 100644 (file)
--- a/src/black/lines.py
+++ b/src/black/lines.py
@@ -520,7 +520,8 @@ class EmptyLineTracker:
                 and (self.semantic_leading_comment is None or before)
             ):
                 self.semantic_leading_comment = block
                 and (self.semantic_leading_comment is None or before)
             ):
                 self.semantic_leading_comment = block

-        elif not current_line.is_decorator:
+        # `or before` means this decorator already has an empty line before
+        elif not current_line.is_decorator or before:

             self.semantic_leading_comment = None
 
         self.previous_line = current_line
             self.semantic_leading_comment = None
 
         self.previous_line = current_line