Fix type annotations

Author: markshannon

Tools/cases_generator/generators_common.py

@@ -491,6 +491,7 @@ def _emit_simple(
     ) -> tuple[bool, Token | None, Storage]:
         local_stores = set(uop.local_stores)
         reachable = True
+        tkn = stmt.contents[-1]
         try:
             if stmt in uop.properties.escaping_calls:
                 escape = uop.properties.escaping_calls[stmt]
@@ -579,6 +580,7 @@ def _emit_if(
             self.out.emit(tkn)
         if_storage = storage.copy()
         try:
+            rbrace: Token | None = stmt.if_
             reachable, rbrace, if_storage = self._emit_stmt(stmt.body, uop, if_storage, inst)
             if stmt.else_ is not None:
                 assert rbrace is not None

Tools/cases_generator/parsing.py

@@ -354,8 +354,8 @@ def label_def(self) -> LabelDef | None:
             if self.expect(lx.LPAREN):
                 if tkn := self.expect(lx.IDENTIFIER):
                     if self.expect(lx.RPAREN):
-                        if block := self.block():
-                            return LabelDef(tkn.text, spilled, block)
+                        block = self.block()
+                        return LabelDef(tkn.text, spilled, block)
         return None
 
     @contextual