Solution ready for publishing: update main.py

main.py

@@ -107,7 +107,7 @@ builder.add_conditional_edges(
 builder.add_conditional_edges(
     "rewrite",
     lambda state: "reflect" if state["round"] < state["max_rounds"] else END,
-    {"reflect": "reflect", "END": END}
+    {"reflect": "reflect", END: END}
 )
 
 graph = builder.compile()