From 0e329025bf797ea2de39d8da05b9ae03c48750f9 Mon Sep 17 00:00:00 2001 From: Tim Blakely Date: Tue, 16 Sep 2025 12:13:22 -0700 Subject: [PATCH] Add default roundtrip encoder/decoders for dataclasses_json for: - connectomics.common.file.Path - np.in64 PiperOrigin-RevId: 807795051 --- ffn/inference/executor.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/ffn/inference/executor.py b/ffn/inference/executor.py index c0461c1..df78185 100644 --- a/ffn/inference/executor.py +++ b/ffn/inference/executor.py @@ -294,9 +294,14 @@ def _run_executor(self): self.active_clients += 1 logging.info('client %d starting', client_id) else: - self.registered_clients.remove(-client_id - 1) - logging.info('client %d terminating', -client_id - 1) - self.active_clients -= 1 + try: + self.registered_clients.remove(-client_id - 1) + logging.info('client %d terminating', -client_id - 1) + self.active_clients -= 1 + except KeyError: + logging.warning( + 'client %d not known or already terminated', -client_id - 1 + ) else: client_id, seed, image, fetches = data l = len(ready)