yzt15806542928's picture
Upload folder using huggingface_hub
f1d3656 verified
Raw
History Blame Contribute Delete
685 Bytes
import logging
import pathlib
import os
class RelPathFormatter(logging.Formatter):
def __init__(self, fmt, datefmt=None):
super().__init__(fmt, datefmt)
self.root_path = pathlib.Path(__file__).parent.parent.parent.resolve()
def format(self, record):
# Replace the full pathname with the relative path
record.pathname = os.path.relpath(record.pathname, self.root_path)
return super().format(record)
logger = logging.getLogger('atmorep')
logger.setLevel(logging.DEBUG)
ch = logging.StreamHandler()
formatter = RelPathFormatter('%(pathname)s:%(lineno)d : %(levelname)-8s : %(message)s')
ch.setFormatter(formatter)
logger.handlers.clear()
logger.addHandler(ch)