Use executors in WorkerManagerClient lock call

Flared · Aug 23, 2024 · eaa02bb · eaa02bb
1 parent b601439
commit eaa02bb
Showing 1 changed file with 7 additions and 1 deletion.
diff --git 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)