From 4ffae6065f1c7fe4dc00fed21ee9a584cbbb14b3 Mon Sep 17 00:00:00 2001 From: vbarda Date: Wed, 14 May 2025 21:35:25 -0400 Subject: [PATCH] lint + update --- libs/langgraph/langgraph/pregel/draw.py | 4 ++-- libs/langgraph/tests/test_remote_graph.py | 4 +++- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/libs/langgraph/langgraph/pregel/draw.py b/libs/langgraph/langgraph/pregel/draw.py index 3ece304f6..5421a7bba 100644 --- a/libs/langgraph/langgraph/pregel/draw.py +++ b/libs/langgraph/langgraph/pregel/draw.py @@ -133,8 +133,8 @@ def draw_graph( task.name: { ( w[0], - (task.name, *w) in conditionals, - conditionals.get((task.name, *w)), + (task.name, w[0], w[1] or None) in conditionals, + conditionals.get((task.name, w[0], w[1] or None)), ) for w in task.writes } diff --git a/libs/langgraph/tests/test_remote_graph.py b/libs/langgraph/tests/test_remote_graph.py index 1358e6f25..82e5d27f1 100644 --- a/libs/langgraph/tests/test_remote_graph.py +++ b/libs/langgraph/tests/test_remote_graph.py @@ -863,7 +863,9 @@ async def test_ainvoke(): assert result == {"messages": [{"type": "human", "content": "world"}]} -@pytest.mark.skip("Unskip this test to manually test the LangGraph Platform integration") +@pytest.mark.skip( + "Unskip this test to manually test the LangGraph Platform integration" +) @pytest.mark.anyio async def test_langgraph_cloud_integration(): from langgraph_sdk.client import get_client, get_sync_client