def METHOD_NAME(self, callback):
    """Register a callback for comm_msg
    Will be called with the `data` of any comm_msg messages.
    Call `on_msg(None)` to disable an existing callback.
    """
    self._msg_callback = callback