Fix stream_mode=messages/custom buffering for sync calls with single node

2026-09-13 13:17:52 +02:00 · 2024-09-25 10:38:06 -07:00
parent 042653207f
commit 67d907eb1c
1 changed files with 2 additions and 2 deletions
@@ -49,8 +49,8 @@ class PregelRunner:
        tasks = tuple(tasks)
        # give control back to the caller
        yield
-        # fast path if single task with no timeout
-        if len(tasks) == 1 and timeout is None:
+        # fast path if single task with no timeout and no waiter
+        if len(tasks) == 1 and timeout is None and get_waiter is None:
            t = tasks[0]
            try:
                run_with_retry(t, retry_policy)