| 354 | |
| 355 | # Intercept default logging and transform to loguru |
| 356 | class InterceptHandler(logging.Handler): |
| 357 | def emit(self, record: logging.LogRecord) -> None: |
| 358 | # Get corresponding Loguru level if it exists. |
| 359 | level: str | int |
| 360 | try: |
| 361 | level = logger.level(record.levelname).name |
| 362 | except ValueError: |
| 363 | level = record.levelno |
| 364 | |
| 365 | # Find caller from where originated the logged message. |
| 366 | frame, depth = inspect.currentframe(), 0 |
| 367 | while frame and (depth == 0 or frame.f_code.co_filename == logging.__file__): |
| 368 | frame = frame.f_back |
| 369 | depth += 1 |
| 370 | |
| 371 | logger.opt(depth=depth, exception=record.exc_info).log(level, record.getMessage()) |
| 372 | |
| 373 | logging.basicConfig(handlers=[InterceptHandler()], level=log_level, force=True) |
| 374 | transformers.utils.logging.disable_default_handler() |