| 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() |