DataCraft-AI
diff --git a/‎makefile‎ ‎Makefile‎makefile renamed to Makefile b/‎makefile‎ ‎Makefile‎makefile renamed to Makefile
diff --git a/‎patterns/behavioral/mediator.py‎
Lines changed: 7 additions & 5 deletions b/‎patterns/behavioral/mediator.py‎
Lines changed: 7 additions & 5 deletions
@@ -8,25 +8,27 @@
 Encapsulates how a set of objects interact.
 """
 
+from __future__ import annotations
+
 
 class ChatRoom:
     """Mediator class"""
 
-    def display_message(self, user, message):
-        print("[{} says]: {}".format(user, message))
+    def display_message(self, user: User, message: str) -> None:
+        print(f"[{user} says]: {message}")
 
 
 class User:
     """A class whose instances want to interact with each other"""
 
-    def __init__(self, name):
+    def __init__(self, name: str) -> None:
         self.name = name
         self.chat_room = ChatRoom()
 
-    def say(self, message):
+    def say(self, message: str) -> None:
         self.chat_room.display_message(self, message)
 
-    def __str__(self):
+    def __str__(self) -> str:
         return self.name