From 52467686da18e61a20c66ae00515914946e67944 Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Tue, 3 Dec 2024 15:29:10 -0800 Subject: [PATCH] Update --- libs/scheduler-kafka/tests/test_subgraph.py | 6 ++++++ libs/scheduler-kafka/tests/test_subgraph_sync.py | 6 ++++++ 2 files changed, 12 insertions(+) diff --git a/libs/scheduler-kafka/tests/test_subgraph.py b/libs/scheduler-kafka/tests/test_subgraph.py index ebaaea5809..6ff2c435b1 100644 --- a/libs/scheduler-kafka/tests/test_subgraph.py +++ b/libs/scheduler-kafka/tests/test_subgraph.py @@ -191,6 +191,7 @@ async def test_subgraph_w_interrupt( "__pregel_delegate": False, "__pregel_read": None, "__pregel_send": None, + "__pregel_call": None, "__pregel_ensure_latest": True, "__pregel_dedupe_tasks": True, "__pregel_resuming": False, @@ -256,6 +257,7 @@ async def test_subgraph_w_interrupt( "__pregel_delegate": False, "__pregel_read": None, "__pregel_send": None, + "__pregel_call": None, "__pregel_ensure_latest": True, "__pregel_dedupe_tasks": True, "__pregel_resuming": False, @@ -351,6 +353,7 @@ async def test_subgraph_w_interrupt( "__pregel_delegate": False, "__pregel_read": None, "__pregel_send": None, + "__pregel_call": None, "__pregel_ensure_latest": True, "__pregel_dedupe_tasks": True, "__pregel_resuming": False, @@ -456,6 +459,7 @@ async def test_subgraph_w_interrupt( "__pregel_delegate": False, "__pregel_read": None, "__pregel_send": None, + "__pregel_call": None, "__pregel_ensure_latest": True, "__pregel_dedupe_tasks": True, "__pregel_resuming": True, @@ -516,6 +520,7 @@ async def test_subgraph_w_interrupt( "__pregel_delegate": False, "__pregel_read": None, "__pregel_send": None, + "__pregel_call": None, "__pregel_ensure_latest": True, "__pregel_dedupe_tasks": True, "__pregel_resuming": True, @@ -632,6 +637,7 @@ async def test_subgraph_w_interrupt( "__pregel_delegate": False, "__pregel_read": None, "__pregel_send": None, + "__pregel_call": None, "__pregel_ensure_latest": True, "__pregel_dedupe_tasks": True, "__pregel_resuming": True, diff --git a/libs/scheduler-kafka/tests/test_subgraph_sync.py b/libs/scheduler-kafka/tests/test_subgraph_sync.py index 75b9d6e730..91c377010b 100644 --- a/libs/scheduler-kafka/tests/test_subgraph_sync.py +++ b/libs/scheduler-kafka/tests/test_subgraph_sync.py @@ -190,6 +190,7 @@ def test_subgraph_w_interrupt( "__pregel_delegate": False, "__pregel_read": None, "__pregel_send": None, + "__pregel_call": None, "__pregel_ensure_latest": True, "__pregel_dedupe_tasks": True, "__pregel_resuming": False, @@ -255,6 +256,7 @@ def test_subgraph_w_interrupt( "__pregel_delegate": False, "__pregel_read": None, "__pregel_send": None, + "__pregel_call": None, "__pregel_ensure_latest": True, "__pregel_store": None, "__pregel_dedupe_tasks": True, @@ -350,6 +352,7 @@ def test_subgraph_w_interrupt( "__pregel_delegate": False, "__pregel_read": None, "__pregel_send": None, + "__pregel_call": None, "__pregel_ensure_latest": True, "__pregel_dedupe_tasks": True, "__pregel_store": None, @@ -454,6 +457,7 @@ def test_subgraph_w_interrupt( "__pregel_delegate": False, "__pregel_read": None, "__pregel_send": None, + "__pregel_call": None, "__pregel_ensure_latest": True, "__pregel_dedupe_tasks": True, "__pregel_store": None, @@ -514,6 +518,7 @@ def test_subgraph_w_interrupt( "__pregel_delegate": False, "__pregel_read": None, "__pregel_send": None, + "__pregel_call": None, "__pregel_ensure_latest": True, "__pregel_dedupe_tasks": True, "__pregel_store": None, @@ -630,6 +635,7 @@ def test_subgraph_w_interrupt( "__pregel_delegate": False, "__pregel_read": None, "__pregel_send": None, + "__pregel_call": None, "__pregel_ensure_latest": True, "__pregel_dedupe_tasks": True, "__pregel_resuming": True,