File size: 437 Bytes
da50cbf | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | import logging
class TqdmToLogger(object):
"""File-like object to redirect tqdm output to a logger."""
def __init__(self, logger, level=logging.INFO):
self.logger = logger
self.level = level
def write(self, buf):
for line in buf.rstrip().splitlines():
self.logger.log(self.level, line)
def flush(self):
for handler in self.logger.logger.handlers:
handler.flush() |