Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
21 changes: 0 additions & 21 deletions gittensor/utils/config.py
Original file line number Diff line number Diff line change
Expand Up @@ -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."""
Expand All @@ -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):
"""
Expand Down Expand Up @@ -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,
Expand Down
35 changes: 0 additions & 35 deletions gittensor/utils/logging.py
Original file line number Diff line number Diff line change
@@ -1,45 +1,10 @@
import logging
import os
from logging.handlers import RotatingFileHandler
from typing import TYPE_CHECKING, List, Optional

import bittensor as bt

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'],
Expand Down
Loading