gh-144016: Fix bad stack assert in the JIT optimizer (GH-144019)

diff --git a/Python/optimizer_analysis.c b/Python/optimizer_analysis.c
index c6a1ae60a317fa4fa3f01070455ed2372893b0e4..6c381ab184fd855c9a457f19843dabe18fdacdb5 100644 (file)
--- a/Python/optimizer_analysis.c
+++ b/Python/optimizer_analysis.c
@@ -507,7 +507,7 @@ optimize_uops(
             *(ctx->out_buffer.next++) = *this_instr;
         }
         assert(ctx->frame != NULL);
-        if (!CURRENT_FRAME_IS_INIT_SHIM()) {
+        if (!CURRENT_FRAME_IS_INIT_SHIM() && !ctx->done) {
             DPRINTF(3, " stack_level %d\n", STACK_LEVEL());
             ctx->frame->stack_pointer = stack_pointer;
             assert(STACK_LEVEL() >= 0);