Skip to content

Commit

Permalink
Logging: Adds _deinitialize_bt_logger to remove from queue
Browse files Browse the repository at this point in the history
  • Loading branch information
Ibraheem Muhammad Nadeem authored and Ibraheem Muhammad Nadeem committed Apr 17, 2024
1 parent a497f65 commit c24a45a
Showing 1 changed file with 13 additions and 2 deletions.
15 changes: 13 additions & 2 deletions bittensor/btlogging/loggingmachine.py
Original file line number Diff line number Diff line change
Expand Up @@ -89,7 +89,7 @@ def __init__(self, config: bittensor.config, name: str = BITTENSOR_LOGGER_NAME):
self._listener = self._create_and_start_listener(self._handlers)

# set up all the loggers
self._logger = self._initialize_bt_logger(name, config)
self._logger = self._initialize_bt_logger(name)
self.disable_third_party_loggers()

def _configure_handlers(self, config) -> list[stdlogging.Handler]:
Expand Down Expand Up @@ -149,7 +149,7 @@ def get_queue(self):
"""
return self._queue

def _initialize_bt_logger(self, name, config):
def _initialize_bt_logger(self, name):
"""
Initialize logging for bittensor.
Expand All @@ -161,6 +161,17 @@ def _initialize_bt_logger(self, name, config):
queue_handler = QueueHandler(self._queue)
logger.addHandler(queue_handler)
return logger

def _deinitialize_bt_logger(self, name):
"""
Find the logger by name and remove the
queue handler associated with it.
"""
logger = stdlogging.getLogger(name)
for handler in list(logger.handlers):
if isinstance(handler, QueueHandler):
logger.removeHandler(handler)
return logger

def _create_file_handler(self, logfile: str):
file_handler = RotatingFileHandler(
Expand Down

0 comments on commit c24a45a

Please sign in to comment.