refactor: move signal_handler from parallel/ to top-level for broader reuse

2026-07-28 10:36:17 +08:00
parent 9f7cf50c56
commit 39f84f3b4c
4 changed files with 3 additions and 3 deletions
@@ -10,7 +10,7 @@ from torch.utils.data import Dataset

 from astrai.config import TrainConfig
 from astrai.model.transformer import AutoRegressiveLM
-from astrai.parallel.signal_handler import register_signal_handlers
+from astrai.signal_handler import register_signal_handlers
 from astrai.trainer import Trainer
 from astrai.trainer.schedule import SchedulerFactory
 from astrai.trainer.train_context import TrainContext