mirror of
https://github.com/twitter/the-algorithm-ml.git
synced 2024-11-17 21:49:21 +01:00
27 lines
763 B
Python
27 lines
763 B
Python
|
"""Sets up logging through absl for training usage.
|
||
|
|
|
||
|
|
- Redirects logging to sys.stdout so that severity levels in GCP Stackdriver are accurate.
|
||
|
|
|
||
|
|
Usage:
|
||
|
|
>>> from twitter.ml.logging.absl_logging import logging
|
||
|
|
>>> logging.info(f"Properly logged as INFO level in GCP Stackdriver.")
|
||
|
|
|
||
|
|
"""
|
||
|
|
import logging as py_logging
|
||
|
|
import sys
|
||
|
|
|
||
|
|
from absl import logging as logging
|
||
|
|
|
||
|
|
|
||
|
|
def setup_absl_logging():
|
||
|
|
"""Make sure that absl logging pushes to stdout rather than stderr."""
|
||
|
|
logging.get_absl_handler().python_handler.stream = sys.stdout
|
||
|
|
formatter = py_logging.Formatter(
|
||
|
|
fmt="[%(module)s.%(funcName)s:%(lineno)s - %(levelname)s] %(message)s"
|
||
|
|
)
|
||
|
|
logging.get_absl_handler().setFormatter(formatter)
|
||
|
|
logging.set_verbosity(logging.INFO)
|
||
|
|
|
||
|
|
|
||
|
|
setup_absl_logging()
|