Agent: Add docstrings to notify_relay_on_propagation

monkey/infection_monkey/agent_event_handlers/notify_relay_on_propagation.py

@@ -8,10 +8,22 @@ logger = logging.getLogger(__name__)
 
 
 class notify_relay_on_propagation:
+    """
+    Notifies a TCPRelay of potential relay users if propagation is successful
+    """
+
     def __init__(self, tcp_relay: Optional[TCPRelay]):
+        """
+        :param tcp_relay: A TCPRelay to notify on successful propagation
+        """
         self._tcp_relay = tcp_relay
 
     def __call__(self, event: PropagationEvent):
+        """
+        Notify a TCPRelay of potential relay users if propagation is successful
+
+        :param event: A `PropagationEvent`
+        """
         if self._tcp_relay is None:
             return