From e8999c23e3c6f913aa2ab22711fa46f5795714f3 Mon Sep 17 00:00:00 2001
From: "sdk-sentinel-publisher[bot]"
 <314421413+sdk-sentinel-publisher[bot]@users.noreply.github.com>
Date: Sun, 16 Aug 2026 00:43:08 +0000
Subject: [PATCH] Fix CI flake detected by SDK Sentinel (python)

---
 tests/worker/test_workflow.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/tests/worker/test_workflow.py b/tests/worker/test_workflow.py
index cca6e779d..1f7079190 100644
--- a/tests/worker/test_workflow.py
+++ b/tests/worker/test_workflow.py
@@ -6845,6 +6845,14 @@ async def _run_workflow_and_get_warning(self) -> bool:
             id=workflow_id,
             task_queue=task_queue,
         )
+        # Keep the update on the run that receives it so continue-as-new cannot
+        # make a completion poll look for the update on the successor run.
+        assert handle.result_run_id
+        handle = self.client.get_workflow_handle(
+            workflow_id,
+            run_id=handle.result_run_id,
+            first_execution_run_id=handle.first_execution_run_id,
+        )
         if self.workflow_termination_type == "-cancellation-":
             await handle.cancel()