File size: 554 Bytes
38136ed | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | """Utility helpers."""
import logging
import sys
def setup_logging(level=logging.INFO):
logging.basicConfig(
level=level,
format="%(asctime)s | %(levelname)s | %(name)s | %(message)s",
handlers=[logging.StreamHandler(sys.stdout)],
force=True,
)
# Quiet HuggingFace
logging.getLogger("transformers").setLevel(logging.WARNING)
logging.getLogger("datasets").setLevel(logging.WARNING)
logging.getLogger("urllib3").setLevel(logging.WARNING)
logging.getLogger("filelock").setLevel(logging.WARNING)
|