From a910a1a3413447a3fe5a88280bd7dbe5e60bd82a Mon Sep 17 00:00:00 2001 From: Vadym Barda Date: Mon, 27 Jan 2025 16:06:31 -0500 Subject: [PATCH] langgraph: actually fix flaky test (#3219) --- libs/langgraph/tests/test_pregel.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/libs/langgraph/tests/test_pregel.py b/libs/langgraph/tests/test_pregel.py index e0a4be241..ed1b64956 100644 --- a/libs/langgraph/tests/test_pregel.py +++ b/libs/langgraph/tests/test_pregel.py @@ -6300,9 +6300,9 @@ def test_named_tasks_functional() -> None: @entrypoint() def workflow(inputs: dict) -> dict: - fut_foo = foo(inputs) + foo_result = foo(inputs).result() other_foo(inputs).result() - fut_bar = bar(fut_foo.result()) + fut_bar = bar(foo_result) fut_baz = baz_task(fut_bar.result()) fut_custom_baz = custom_baz_task(fut_baz.result()) fut_qux = qux_task(fut_custom_baz.result())