fix mypy

JelleZijlstra · May 3, 2023 · e9678f4 · e9678f4
1 parent 1001d92
commit e9678f4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/blib2to3/pgen2/tokenize.py b/src/blib2to3/pgen2/tokenize.py
@@ -598,12 +598,13 @@ def generate_tokens(
                 ):
                     if token[-1] == "\n":  # continued string
                         strstart = (lnum, start)
-                        endprog = (
+                        maybe_endprog = (
                             endprogs[initial]
                             or endprogs.get(token[1])
                             or endprogs.get(token[2])
                         )
-                        assert endprog is not None, f"endprog not found for {token}"
+                        assert maybe_endprog is not None, f"endprog not found for {token}"
+                        endprog = maybe_endprog
                         contstr, needcont = line[start:], 1
                         contline = line
                         break