This commit is contained in:
Sydney Runkle
2026-02-27 15:46:57 -05:00
parent 71c0500caa
commit 4c3c30aaf4
+8 -8
View File
@@ -3105,11 +3105,11 @@ class Pregel(
durability=durability,
**kwargs,
):
# when subgraphs=True with a single stream_mode, stream()
# yields (namespace, payload) tuples — unwrap them
if subgraphs and isinstance(chunk, tuple):
chunk = chunk[-1]
if stream_mode == "values":
# when subgraphs=True with a single stream_mode, stream()
# yields (namespace, payload) tuples — unwrap them
if subgraphs and isinstance(chunk, tuple):
chunk = chunk[-1]
latest = chunk
if (
isinstance(chunk, dict)
@@ -3191,11 +3191,11 @@ class Pregel(
durability=durability,
**kwargs,
):
# when subgraphs=True with a single stream_mode, astream()
# yields (namespace, payload) tuples — unwrap them
if subgraphs and isinstance(chunk, tuple):
chunk = chunk[-1]
if stream_mode == "values":
# when subgraphs=True with a single stream_mode, astream()
# yields (namespace, payload) tuples — unwrap them
if subgraphs and isinstance(chunk, tuple):
chunk = chunk[-1]
latest = chunk
if (
isinstance(chunk, dict)