From 27cf03a444bd818a837626d9730bf0bc3bdde6fe Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Sat, 24 Feb 2024 19:46:29 -0800 Subject: [PATCH] Lint --- langgraph/graph/graph.py | 2 +- langgraph/graph/state.py | 5 ++--- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/langgraph/graph/graph.py b/langgraph/graph/graph.py index 53331d2af..b4ce617a8 100644 --- a/langgraph/graph/graph.py +++ b/langgraph/graph/graph.py @@ -14,7 +14,7 @@ from langchain_core.runnables.graph import Graph as RunnableGraph from langgraph.channels.ephemeral_value import EphemeralValue from langgraph.checkpoint import BaseCheckpointSaver -from langgraph.pregel import Channel, Pregel, StateSnapshot +from langgraph.pregel import Channel, Pregel logger = logging.getLogger(__name__) diff --git a/langgraph/graph/state.py b/langgraph/graph/state.py index b1f048074..5c6b1d556 100644 --- a/langgraph/graph/state.py +++ b/langgraph/graph/state.py @@ -1,11 +1,10 @@ from collections import defaultdict from functools import partial from inspect import signature -from typing import Any, Optional, Sequence, Type, Union +from typing import Any, Optional, Sequence, Type from langchain_core.runnables import RunnableLambda from langchain_core.runnables.base import RunnableLike -from langchain_core.runnables.config import RunnableConfig from langgraph.channels.any_value import AnyValue from langgraph.channels.base import BaseChannel, InvalidUpdateError @@ -14,7 +13,7 @@ from langgraph.channels.ephemeral_value import EphemeralValue from langgraph.channels.last_value import LastValue from langgraph.checkpoint import BaseCheckpointSaver from langgraph.graph.graph import END, START, CompiledGraph, Graph -from langgraph.pregel import Channel, StateSnapshot +from langgraph.pregel import Channel from langgraph.pregel.read import ChannelInvoke from langgraph.pregel.write import SKIP_WRITE, ChannelWrite, ChannelWriteEntry