Replace get_stdout() with SAFE_STDOUT

2023-12-30 18:45:04 +09:00
parent 6e8bcc2219
commit a9652979cc
10 changed files with 24 additions and 24 deletions
--- a/resample.py
+++ b/resample.py
@@ -9,7 +9,7 @@ from tqdm import tqdm

 from config import config
 from tools.log import logger
-from tools.stdout_wrapper import get_stdout
+from tools.stdout_wrapper import SAFE_STDOUT


 def normalize_audio(data, sr):
@@ -88,7 +88,7 @@ if __name__ == "__main__":

    pool = Pool(processes=processes)
    for _ in tqdm(
-        pool.imap_unordered(process, tasks), file=get_stdout(), total=len(tasks)
+        pool.imap_unordered(process, tasks), file=SAFE_STDOUT, total=len(tasks)
    ):
        pass