Ignore colon-after-lambda in compound statement rules (#2771)

astral-sh · Feb 11, 2023 · 43b7ee2 · 43b7ee2
1 parent 77099dc
commit 43b7ee2
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/crates/ruff/resources/test/fixtures/pycodestyle/E70.py b/crates/ruff/resources/test/fixtures/pycodestyle/E70.py
@@ -44,3 +44,5 @@ class C: pass
 #:
 if a := 1:
     pass
+#:
+func = lambda x: x** 2 if cond else lambda x:x
diff --git a/crates/ruff/src/rules/pycodestyle/rules/compound_statements.rs b/crates/ruff/src/rules/pycodestyle/rules/compound_statements.rs
@@ -182,6 +182,21 @@ pub fn compound_statements(lxr: &[LexResult]) -> Vec<Diagnostic> {
         }
 
         match tok {
+            Tok::Lambda => {
+                // Reset.
+                def = None;
+                colon = None;
+                class = None;
+                elif = None;
+                else_ = None;
+                except = None;
+                finally = None;
+                for_ = None;
+                if_ = None;
+                try_ = None;
+                while_ = None;
+                with = None;
+            }
             Tok::If => {
                 if_ = Some((start, end));
             }
-Original file line number
+Diff line change
@@ Expand Up / @@ -44,3 +44,5 @@ class C: pass @@
     #:
     if a := 1:
         pass
+    #:
+    func = lambda x: x** 2 if cond else lambda x:x