diff --git a/libs/langgraph/tests/test_pregel.py b/libs/langgraph/tests/test_pregel.py index 7702d79e2..157a02cc7 100644 --- a/libs/langgraph/tests/test_pregel.py +++ b/libs/langgraph/tests/test_pregel.py @@ -8564,7 +8564,7 @@ def test_subgraph_resume_reexecutes_from_valid_checkpoint( outer_model_calls = 0 - def outer_model(state: OuterState) -> Command | None: + def outer_model(state: OuterState) -> Optional[Command]: nonlocal outer_model_calls outer_model_calls += 1 if "output" in state: diff --git a/libs/langgraph/tests/test_pregel_async.py b/libs/langgraph/tests/test_pregel_async.py index b560a82eb..1bd8fa143 100644 --- a/libs/langgraph/tests/test_pregel_async.py +++ b/libs/langgraph/tests/test_pregel_async.py @@ -9259,7 +9259,7 @@ async def test_subgraph_resume_reexecutes_from_valid_checkpoint( outer_model_calls = 0 - async def outer_model(state: OuterState) -> Command | None: + async def outer_model(state: OuterState) -> Optional[Command]: nonlocal outer_model_calls outer_model_calls += 1 if "output" in state: