diff --git a/gittensor/utils/config.py b/gittensor/utils/config.py index 2e28c2b5..9f16dcf9 100644 --- a/gittensor/utils/config.py +++ b/gittensor/utils/config.py @@ -22,8 +22,6 @@ import bittensor as bt -from gittensor.utils.logging import setup_events_logger - def check_config(cls, config: Any): r"""Checks/validates the config namespace object.""" @@ -43,11 +41,6 @@ def check_config(cls, config: Any): if not os.path.exists(config.neuron.full_path): os.makedirs(config.neuron.full_path, exist_ok=True) - if not config.neuron.dont_save_events: - # Add custom event logger for the events. - events_logger = setup_events_logger(config.neuron.full_path, config.neuron.events_retention_size) - bt.logging.register_primary_logger(events_logger.name) - def add_args(cls, parser): """ @@ -77,20 +70,6 @@ def add_args(cls, parser): default=False, ) - parser.add_argument( - '--neuron.events_retention_size', - type=str, - help='Events retention size.', - default=2 * 1024 * 1024 * 1024, # 2 GB - ) - - parser.add_argument( - '--neuron.dont_save_events', - action='store_true', - help='If set, we dont save events to a log file.', - default=False, - ) - parser.add_argument( '--wandb.project', type=str, diff --git a/gittensor/utils/logging.py b/gittensor/utils/logging.py index 0967b4c9..5ed94b6c 100644 --- a/gittensor/utils/logging.py +++ b/gittensor/utils/logging.py @@ -1,6 +1,3 @@ -import logging -import os -from logging.handlers import RotatingFileHandler from typing import TYPE_CHECKING, List, Optional import bittensor as bt @@ -8,38 +5,6 @@ if TYPE_CHECKING: from gittensor.classes import FileScoreResult, ScoreBreakdown -EVENTS_LEVEL_NUM = 38 -DEFAULT_LOG_BACKUP_COUNT = 10 - - -def setup_events_logger(full_path, events_retention_size): - logging.addLevelName(EVENTS_LEVEL_NUM, 'EVENT') - - logger = logging.getLogger('event') - logger.setLevel(EVENTS_LEVEL_NUM) - - def event(self, message, *args, **kws): - if self.isEnabledFor(EVENTS_LEVEL_NUM): - self._log(EVENTS_LEVEL_NUM, message, args, **kws) - - logging.Logger.event = event # type: ignore[attr-defined] - - formatter = logging.Formatter( - '%(asctime)s | %(levelname)s | %(message)s', - datefmt='%Y-%m-%d %H:%M:%S', - ) - - file_handler = RotatingFileHandler( - os.path.join(full_path, 'events.log'), - maxBytes=events_retention_size, - backupCount=DEFAULT_LOG_BACKUP_COUNT, - ) - file_handler.setFormatter(formatter) - file_handler.setLevel(EVENTS_LEVEL_NUM) - logger.addHandler(file_handler) - - return logger - def log_scoring_results( file_results: List['FileScoreResult'],