Fix: colab stdout error, so replace with stdout wrapper

tools/log.py

@@ -2,8 +2,8 @@
 logger封装
 """
 from loguru import logger
-import sys
 
+from .stdout_wrapper import get_stdout
 
 # 移除所有默认的处理器
 logger.remove()
@@ -13,4 +13,4 @@ log_format = (
     "<g>{time:MM-DD HH:mm:ss}</g> |<lvl>{level:^8}</lvl>| {file}:{line} | {message}"
 )
 
-logger.add(sys.stdout, format=log_format, backtrace=True, diagnose=True)
+logger.add(get_stdout(), format=log_format, backtrace=True, diagnose=True)

tools/subprocess_utils.py

@@ -2,6 +2,7 @@ import subprocess
 import sys
 
 from .log import logger
+from .stdout_wrapper import get_stdout
 
 python = sys.executable
 
@@ -10,7 +11,7 @@ def run_script_with_log(cmd: list[str]) -> tuple[bool, str]:
     logger.info(f"Running: {' '.join(cmd)}")
     result = subprocess.run(
         [python] + cmd,
-        stdout=sys.stdout,
+        stdout=get_stdout(),  # type: ignore
         stderr=subprocess.PIPE,
         text=True,
     )