Method close

axlearn/cloud/common/event_queue.py:186–195

Closes the RabbitMQ connection.

(self)

            raise e

    def close(self):
        """Closes the RabbitMQ connection."""
        if self._connection and not self._connection.is_closed:
            self._connection.close()
            logging.info("Closed RabbitMQ connection.")
        if self._channel and not self._channel.is_closed:
            self._channel.close()
            logging.info("Closed RabbitMQ channel.")
        self._connection = None
        self._channel = None

    def publish(self, event: Event):
        """Publishes an event to the queue."""