From 144ee31546e936738ac2e14577f68792fd9f33fa Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Wed, 15 Jan 2025 13:43:14 -0800 Subject: [PATCH] Lint --- libs/scheduler-kafka/tests/test_subgraph.py | 2 +- libs/scheduler-kafka/tests/test_subgraph_sync.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/libs/scheduler-kafka/tests/test_subgraph.py b/libs/scheduler-kafka/tests/test_subgraph.py index 88042b28a..55303cba0 100644 --- a/libs/scheduler-kafka/tests/test_subgraph.py +++ b/libs/scheduler-kafka/tests/test_subgraph.py @@ -15,7 +15,7 @@ from langgraph.graph.state import StateGraph from langgraph.pregel import Pregel from langgraph.scheduler.kafka import serde from langgraph.scheduler.kafka.types import MessageToOrchestrator, Topics -from tests.any import AnyDict, AnyList +from tests.any import AnyDict from tests.drain import drain_topics_async from tests.messages import _AnyIdAIMessage, _AnyIdHumanMessage diff --git a/libs/scheduler-kafka/tests/test_subgraph_sync.py b/libs/scheduler-kafka/tests/test_subgraph_sync.py index 30b70720f..a67919dda 100644 --- a/libs/scheduler-kafka/tests/test_subgraph_sync.py +++ b/libs/scheduler-kafka/tests/test_subgraph_sync.py @@ -15,7 +15,7 @@ from langgraph.pregel import Pregel from langgraph.scheduler.kafka import serde from langgraph.scheduler.kafka.default_sync import DefaultProducer from langgraph.scheduler.kafka.types import MessageToOrchestrator, Topics -from tests.any import AnyDict, AnyList +from tests.any import AnyDict from tests.drain import drain_topics from tests.messages import _AnyIdAIMessage, _AnyIdHumanMessage