fix: ProgressBar默认输出到stdout

- file参数默认值改为None, 内部用 or sys.stdout 兜底
- 清理inference API中未使用的import (Optional, time, field)
- 删除test_protocol中未使用的ctx变量
This commit is contained in:
2026-05-26 13:27:05 +08:00
parent 94d6e713e9
commit dd1b39f435
5 changed files with 5 additions and 7 deletions
+2 -2
View File
@@ -210,7 +210,7 @@ class ProgressBarCallback(TrainCallback):
"""
def __init__(
self, num_epoch: int, log_interval: int = 100, file: IO[str] = sys.stdout
self, num_epoch: int, log_interval: int = 100, file: Optional[IO[str]] = None
):
self.num_epoch = num_epoch
self.log_interval = log_interval
@@ -223,7 +223,7 @@ class ProgressBarCallback(TrainCallback):
context.dataloader,
desc=f"Epoch {context.epoch + 1}/{self.num_epoch}",
dynamic_ncols=True,
file=self.file,
file=self.file or sys.stdout,
)
@only_on_rank(0)