From 2f8f9e51cefb60add9c9e53aeab898add1099eaa Mon Sep 17 00:00:00 2001
From: Heng Pan <pan@flower.ai>
Date: Sat, 6 Apr 2024 12:07:19 +0100
Subject: [PATCH] save exception to state (#3225)

---
 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 = (