From 5163d3a47aae96c51b0678d20831af70239ff948 Mon Sep 17 00:00:00 2001
From: ViperEkura <3081035982@qq.com>
Date: Mon, 29 Sep 2025 13:08:41 +0800
Subject: [PATCH] =?UTF-8?q?fix(callback):=20=E8=A7=A3=E5=86=B3=E5=BE=AA?=
 =?UTF-8?q?=E7=8E=AF=E5=AF=BC=E5=85=A5=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 khaosz/trainer/callback.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/khaosz/trainer/callback.py b/khaosz/trainer/callback.py
index 4e55311..05ba1ba 100644
--- a/khaosz/trainer/callback.py
+++ b/khaosz/trainer/callback.py
@@ -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: