From 7afdcdbace712a736939773d1ae252d48e05921a 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 --- src/saturn_engine/client/worker_manager.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) 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)