fix(callback): 解决循环导入问题

2025-09-29 13:08:41 +08:00
parent b2f3fefa1b
commit 5163d3a47a
1 changed files with 4 additions and 2 deletions
@@ -1,8 +1,10 @@
 from tqdm import tqdm
 from khaosz.core.parameter import Checkpoint
 from khaosz.trainer.trainer import Trainer
 from torch.nn.utils import clip_grad_norm_
-from typing import cast
+from typing import cast, TYPE_CHECKING
 if TYPE_CHECKING:
    from khaosz.trainer.trainer import Trainer
 class TrainerCallback: