From 065e0ff36066e4232436b866d5f49e968db5486d Mon Sep 17 00:00:00 2001 From: Andrew Nguonly Date: Mon, 13 May 2024 18:34:54 -0700 Subject: [PATCH] Remove unused imports. --- langgraph/managed/few_shot.py | 1 - tests/checkpoint/test_memory.py | 2 -- tests/test_pregel.py | 2 -- 3 files changed, 5 deletions(-) diff --git a/langgraph/managed/few_shot.py b/langgraph/managed/few_shot.py index 4738f0b2e..1e2147006 100644 --- a/langgraph/managed/few_shot.py +++ b/langgraph/managed/few_shot.py @@ -13,7 +13,6 @@ from langchain_core.runnables import RunnableConfig from typing_extensions import Self from langgraph.channels.base import AsyncChannelsManager, ChannelsManager -from langgraph.checkpoint.base import CheckpointTuple from langgraph.managed.base import ManagedValue, V from langgraph.pregel.io import read_channels from langgraph.pregel.types import PregelTaskDescription diff --git a/tests/checkpoint/test_memory.py b/tests/checkpoint/test_memory.py index bbbf0822e..2a4f26e34 100644 --- a/tests/checkpoint/test_memory.py +++ b/tests/checkpoint/test_memory.py @@ -1,5 +1,3 @@ -from typing import AsyncIterator - import pytest from langchain_core.runnables import RunnableConfig diff --git a/tests/test_pregel.py b/tests/test_pregel.py index 723e993d6..7ccc1bd7d 100644 --- a/tests/test_pregel.py +++ b/tests/test_pregel.py @@ -9,7 +9,6 @@ from typing import ( Annotated, Any, Generator, - Iterator, Literal, Optional, Sequence, @@ -26,7 +25,6 @@ from langgraph.channels.binop import BinaryOperatorAggregate from langgraph.channels.context import Context from langgraph.channels.last_value import LastValue from langgraph.channels.topic import Topic -from langgraph.checkpoint.base import CheckpointMetadata, CheckpointTuple from langgraph.checkpoint.sqlite import SqliteSaver from langgraph.errors import InvalidUpdateError from langgraph.graph import END, Graph