diff --git a/libs/scheduler-kafka/tests/test_subgraph.py b/libs/scheduler-kafka/tests/test_subgraph.py index fd2530843..ebaaea580 100644 --- a/libs/scheduler-kafka/tests/test_subgraph.py +++ b/libs/scheduler-kafka/tests/test_subgraph.py @@ -194,8 +194,9 @@ async def test_subgraph_w_interrupt( "__pregel_ensure_latest": True, "__pregel_dedupe_tasks": True, "__pregel_resuming": False, - '__pregel_store': None, + "__pregel_store": None, "__pregel_task_id": history[0].tasks[0].id, + "__pregel_resume_value": None, "checkpoint_id": None, "checkpoint_map": { "": history[0].config["configurable"]["checkpoint_id"] @@ -258,8 +259,9 @@ async def test_subgraph_w_interrupt( "__pregel_ensure_latest": True, "__pregel_dedupe_tasks": True, "__pregel_resuming": False, - '__pregel_store': None, + "__pregel_store": None, "__pregel_task_id": history[0].tasks[0].id, + "__pregel_resume_value": None, "checkpoint_id": c.config["configurable"]["checkpoint_id"], "checkpoint_map": { "": history[0].config["configurable"]["checkpoint_id"] @@ -352,8 +354,9 @@ async def test_subgraph_w_interrupt( "__pregel_ensure_latest": True, "__pregel_dedupe_tasks": True, "__pregel_resuming": False, - '__pregel_store': None, + "__pregel_store": None, "__pregel_task_id": history[0].tasks[0].id, + "__pregel_resume_value": None, "checkpoint_id": c.config["configurable"]["checkpoint_id"], "checkpoint_map": { "": history[0].config["configurable"]["checkpoint_id"] @@ -456,8 +459,9 @@ async def test_subgraph_w_interrupt( "__pregel_ensure_latest": True, "__pregel_dedupe_tasks": True, "__pregel_resuming": True, - '__pregel_store': None, + "__pregel_store": None, "__pregel_task_id": history[1].tasks[0].id, + "__pregel_resume_value": None, "checkpoint_id": None, "checkpoint_map": { "": history[1].config["configurable"]["checkpoint_id"] @@ -515,8 +519,9 @@ async def test_subgraph_w_interrupt( "__pregel_ensure_latest": True, "__pregel_dedupe_tasks": True, "__pregel_resuming": True, - '__pregel_store': None, + "__pregel_store": None, "__pregel_task_id": history[1].tasks[0].id, + "__pregel_resume_value": None, "checkpoint_id": c.config["configurable"]["checkpoint_id"], "checkpoint_map": { "": history[1].config["configurable"]["checkpoint_id"] @@ -630,8 +635,9 @@ async def test_subgraph_w_interrupt( "__pregel_ensure_latest": True, "__pregel_dedupe_tasks": True, "__pregel_resuming": True, - '__pregel_store': None, + "__pregel_store": None, "__pregel_task_id": history[1].tasks[0].id, + "__pregel_resume_value": None, "checkpoint_id": c.config["configurable"]["checkpoint_id"], "checkpoint_map": { "": history[1].config["configurable"]["checkpoint_id"] diff --git a/libs/scheduler-kafka/tests/test_subgraph_sync.py b/libs/scheduler-kafka/tests/test_subgraph_sync.py index 32d0ceea0..75b9d6e73 100644 --- a/libs/scheduler-kafka/tests/test_subgraph_sync.py +++ b/libs/scheduler-kafka/tests/test_subgraph_sync.py @@ -193,8 +193,9 @@ def test_subgraph_w_interrupt( "__pregel_ensure_latest": True, "__pregel_dedupe_tasks": True, "__pregel_resuming": False, - '__pregel_store': None, + "__pregel_store": None, "__pregel_task_id": history[0].tasks[0].id, + "__pregel_resume_value": None, "checkpoint_id": None, "checkpoint_map": { "": history[0].config["configurable"]["checkpoint_id"] @@ -255,10 +256,11 @@ def test_subgraph_w_interrupt( "__pregel_read": None, "__pregel_send": None, "__pregel_ensure_latest": True, - '__pregel_store': None, + "__pregel_store": None, "__pregel_dedupe_tasks": True, "__pregel_resuming": False, "__pregel_task_id": history[0].tasks[0].id, + "__pregel_resume_value": None, "checkpoint_id": c.config["configurable"]["checkpoint_id"], "checkpoint_map": { "": history[0].config["configurable"]["checkpoint_id"] @@ -350,9 +352,10 @@ def test_subgraph_w_interrupt( "__pregel_send": None, "__pregel_ensure_latest": True, "__pregel_dedupe_tasks": True, - '__pregel_store': None, + "__pregel_store": None, "__pregel_resuming": False, "__pregel_task_id": history[0].tasks[0].id, + "__pregel_resume_value": None, "checkpoint_id": c.config["configurable"]["checkpoint_id"], "checkpoint_map": { "": history[0].config["configurable"]["checkpoint_id"] @@ -453,9 +456,10 @@ def test_subgraph_w_interrupt( "__pregel_send": None, "__pregel_ensure_latest": True, "__pregel_dedupe_tasks": True, - '__pregel_store': None, + "__pregel_store": None, "__pregel_resuming": True, "__pregel_task_id": history[1].tasks[0].id, + "__pregel_resume_value": None, "checkpoint_id": None, "checkpoint_map": { "": history[1].config["configurable"]["checkpoint_id"] @@ -512,9 +516,10 @@ def test_subgraph_w_interrupt( "__pregel_send": None, "__pregel_ensure_latest": True, "__pregel_dedupe_tasks": True, - '__pregel_store': None, + "__pregel_store": None, "__pregel_resuming": True, "__pregel_task_id": history[1].tasks[0].id, + "__pregel_resume_value": None, "checkpoint_id": c.config["configurable"]["checkpoint_id"], "checkpoint_map": { "": history[1].config["configurable"]["checkpoint_id"] @@ -628,8 +633,9 @@ def test_subgraph_w_interrupt( "__pregel_ensure_latest": True, "__pregel_dedupe_tasks": True, "__pregel_resuming": True, - '__pregel_store': None, + "__pregel_store": None, "__pregel_task_id": history[1].tasks[0].id, + "__pregel_resume_value": None, "checkpoint_id": c.config["configurable"]["checkpoint_id"], "checkpoint_map": { "": history[1].config["configurable"]["checkpoint_id"]