fixing tests

libs/langgraph/langgraph/pregel/algo.py

@@ -908,7 +908,17 @@ def _scratchpad(
             if not isinstance(task_resume_write, list):
                 task_resume_write = [task_resume_write]
         else:
-            task_resume_write = []
+            # find task-specific resume value
+            for w in pending_writes:
+                if w[0] == task_id and w[1] == RESUME:
+                    task_resume_write = w[2]
+                    if not isinstance(task_resume_write, list):
+                        task_resume_write = [task_resume_write]
+                    break
+            else:
+                task_resume_write = []
+            # clear var
+            del w
     else:
         null_resume_write = None
         task_resume_write = []

libs/langgraph/tests/test_checkpoint_migration.py

@@ -114,7 +114,9 @@ def get_expected_history(*, exc_task_results: int = 0) -> list[StateSnapshot]:
                     else {"answer": "doc1,doc2,doc3,doc4"},
                 ),
             ),
-            interrupts=(
+            interrupts=()
+            if exc_task_results
+            else (
                 Interrupt(
                     value="",
                     resumable=True,

libs/langgraph/tests/test_pregel.py

@@ -6218,7 +6218,7 @@ def test_multiple_interrupts_functional(
         values = []
 
         for idx in [1, 2, 3]:
-            values.extend([double(idx).result(), interrupt({"a": "boo"})])
+            values.extend([double(idx).result(), interrupt({"a": f"boo + {idx}"})])
 
         return {"values": values}