Skip to content
Merged
Show file tree
Hide file tree
Changes from 3 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
7 changes: 0 additions & 7 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
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