From 5e0273886e10180e4557d6e371d4ba39a98d984b Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Tue, 10 Sep 2024 16:02:47 -0700 Subject: [PATCH] Lint --- libs/scheduler-kafka/langgraph/scheduler/kafka/executor.py | 3 ++- libs/scheduler-kafka/langgraph/scheduler/kafka/orchestrator.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/libs/scheduler-kafka/langgraph/scheduler/kafka/executor.py b/libs/scheduler-kafka/langgraph/scheduler/kafka/executor.py index 8aec3ef9e..86812c258 100644 --- a/libs/scheduler-kafka/langgraph/scheduler/kafka/executor.py +++ b/libs/scheduler-kafka/langgraph/scheduler/kafka/executor.py @@ -1,11 +1,12 @@ import asyncio from contextlib import AbstractAsyncContextManager, AsyncExitStack from functools import partial -from typing import Any, Optional, Self, Sequence +from typing import Any, Optional, Sequence import aiokafka import orjson from langchain_core.runnables import RunnableConfig +from typing_extensions import Self import langgraph.scheduler.kafka.serde as serde from langgraph.constants import CONFIG_KEY_DELEGATE, ERROR, NS_END, NS_SEP diff --git a/libs/scheduler-kafka/langgraph/scheduler/kafka/orchestrator.py b/libs/scheduler-kafka/langgraph/scheduler/kafka/orchestrator.py index a3da188e8..c13de9c08 100644 --- a/libs/scheduler-kafka/langgraph/scheduler/kafka/orchestrator.py +++ b/libs/scheduler-kafka/langgraph/scheduler/kafka/orchestrator.py @@ -1,9 +1,10 @@ import asyncio from contextlib import AbstractAsyncContextManager, AsyncExitStack -from typing import Any, Optional, Self +from typing import Any, Optional import aiokafka from langchain_core.runnables import ensure_config +from typing_extensions import Self import langgraph.scheduler.kafka.serde as serde from langgraph.constants import (