Allow lambda (with implicit return) in except* clauses

Author: sterliakov

mypy/semanal.py

@@ -6134,7 +6134,8 @@ def analyze_comp_for_2(self, expr: GeneratorExpr | DictionaryComprehension) -> N
 
     def visit_lambda_expr(self, expr: LambdaExpr) -> None:
         self.analyze_arg_initializers(expr)
-        self.analyze_function_body(expr)
+        with self.inside_except_star_block_set(False, entering_loop=False):
+            self.analyze_function_body(expr)
 
     def visit_conditional_expr(self, expr: ConditionalExpr) -> None:
         expr.if_expr.accept(self)

test-data/unit/check-python311.test

@@ -260,6 +260,24 @@ def foo():
             return  # E: "return" not allowed in except* block
 [builtins fixtures/exception.pyi]
 
+[case testLambdaInExceptStarBlock]
+# flags: --python-version 3.11
+def foo():
+    try:
+        pass
+    except* Exception:
+        x = lambda: 0
+        return lambda: 0  # E: "return" not allowed in except* block
+
+def loop():
+    while True:
+        try:
+            pass
+        except* Exception:
+            x = lambda: 0
+            return lambda: 0  # E: "return" not allowed in except* block
+[builtins fixtures/exception.pyi]
+
 [case testRedefineLocalWithinExceptStarTryClauses]
 # flags: --allow-redefinition
 def fn_str(_: str) -> int: ...