wehub-resource-sync
27 lines
789 B
Python
27 lines
789 B
Python
import logging
|
|
|
|
from ray._common.filters import CoreContextFilter
|
|
from ray._common.formatters import JSONFormatter
|
|
|
|
|
|
def _configure_stdlib_logging():
|
|
"""Configures stdlib root logger to make sure stdlib loggers (created as
|
|
`logging.getLogger(...)`) are using Ray's `JSONFormatter` with Core and Serve
|
|
context filters.
|
|
"""
|
|
|
|
handler = logging.StreamHandler()
|
|
handler.addFilter(CoreContextFilter())
|
|
handler.setFormatter(JSONFormatter())
|
|
|
|
root_logger = logging.getLogger()
|
|
# NOTE: It's crucial we reset all the handlers of the root logger,
|
|
# to make sure that logs aren't emitted twice
|
|
root_logger.handlers = []
|
|
root_logger.addHandler(handler)
|
|
root_logger.setLevel(logging.INFO)
|
|
|
|
|
|
def setup_logging():
|
|
_configure_stdlib_logging()
|