From edcfe76974b755f70d515742de161e1cf3b0b835 Mon Sep 17 00:00:00 2001 From: Olivier Michaud Date: Fri, 23 Aug 2024 08:57:12 -0400 Subject: [PATCH] Use executors in WorkerManagerClient lock call --- pyproject.toml | 2 +- src/saturn_engine/client/worker_manager.py | 8 +++++++- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 55c7dee3..551957eb 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "saturn-engine" -version = "0.5.23" +version = "0.5.22" description = "" readme = "README.md" authors = ["Flare Systems "] diff --git a/src/saturn_engine/client/worker_manager.py b/src/saturn_engine/client/worker_manager.py index f2a22c01..e6e0562f 100644 --- a/src/saturn_engine/client/worker_manager.py +++ b/src/saturn_engine/client/worker_manager.py @@ -49,7 +49,13 @@ def __init__( async def lock(self) -> LockResponse: lock_url = urlcat(self.base_url, "api/lock") - json = asdict(LockInput(worker_id=self.worker_id, selector=self.selector)) + json = asdict( + LockInput( + worker_id=self.worker_id, + selector=self.selector, + executors=self.executors, + ) + ) async with self.http_client.post(lock_url, json=json) as response: return fromdict(await response.json(), LockResponse)