WEBDIENSTE/langgraph
1
0
Fork 0
mirror of https://github.com/langchain-ai/langgraph.git synced 2026-08-21 15:12:26 +02:00
Code Issues Packages Projects Releases Wiki Activity

Lint

Browse Source
This commit is contained in:
Nuno Campos
2025-05-14 11:30:49 -07:00
parent 3acf63a918
commit 3bdb7d09be
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libs/scheduler-kafka/langgraph/scheduler/kafka/executor.py
+6 -4
View File
@@ -221,9 +221,10 @@ class AsyncKafkaExecutor(AbstractAsyncContextManager):
runner = PregelRunner(
submit=weakref.ref(submit),
put_writes=weakref.ref(put_writes),
schedule_task=weakref.WeakMethod(self._schedule_task),
)
async for _ in runner.atick([task], reraise=False):
async for _ in runner.atick(
[task], reraise=False, schedule_task=self._schedule_task
):
pass
else:
# task was not found
@@ -438,9 +439,10 @@ class KafkaExecutor(AbstractContextManager):
runner = PregelRunner(
submit=weakref.ref(submit),
put_writes=weakref.ref(put_writes),
schedule_task=weakref.WeakMethod(self._schedule_task),
)
for _ in runner.tick([task], reraise=False):
for _ in runner.tick(
[task], reraise=False, schedule_task=self._schedule_task
):
pass
else:
# task was not found