From 422f07cada0fa47eab0b9bc43245f9344f9c9e7b Mon Sep 17 00:00:00 2001
From: mohammadnaseri <mohammad@flower.ai>
Date: Fri, 27 Sep 2024 22:40:05 +0100
Subject: [PATCH] Edit

---
 src/py/flwr/common/message.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/py/flwr/common/message.py b/src/py/flwr/common/message.py
index 8a92df4861db..1a49e80fe500 100644
--- a/src/py/flwr/common/message.py
+++ b/src/py/flwr/common/message.py
@@ -362,7 +362,7 @@ def __repr__(self) -> str:
         )
         return f"{self.__class__.__qualname__}({view})"
 
-    def _limit_task_res_ttl(self, message):
+    def _limit_task_res_ttl(self, message: Message) -> None:
         """Limit the TaskRes TTL to not exceed the expiration time of the TaskIns it
         replies to.