Захватить стандартный вывод функции python, чтобы обновить индикатор выполнения TQDM - PullRequest
0 голосов
/ 12 июля 2020

У меня есть функция, которая печатает текст по мере его выполнения. Вывод на печать выглядит следующим образом:

Batch number: 0
Batch number: 1
Batch number: 2

Общее количество строк известно заранее.

Я хотел бы захватить стандартный вывод этой функции, чтобы обновить TQDM индикатор выполнения.

Я видел этот вопрос на SO, который, кажется, соответствует моей проблеме. Ответ выглядит многообещающим, однако мне интересно, как мне создать подкласс StringIO, чтобы инициировать и обновлять индикатор выполнения TQDM.

Заранее благодарим за любую подсказку!

...