From fec78cd6a3f9170d319a9cca7ccc315128b85ab3 Mon Sep 17 00:00:00 2001 From: Heng Pan Date: Fri, 5 Apr 2024 11:12:27 +0100 Subject: [PATCH] save exception to state --- src/py/flwr/common/retry_invoker.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/py/flwr/common/retry_invoker.py b/src/py/flwr/common/retry_invoker.py index 7cec319e7906..d12124b89840 100644 --- a/src/py/flwr/common/retry_invoker.py +++ b/src/py/flwr/common/retry_invoker.py @@ -261,6 +261,7 @@ def try_call_event_handler( try: ret = target(*args, **kwargs) except self.recoverable_exceptions as err: + state.exception = err # Check if giveup event should be triggered max_tries_exceeded = try_cnt == self.max_tries max_time_exceeded = (