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
@@ -12,7 +12,7 @@ import torch
 import torch.distributed as dist
 import torch.multiprocessing as mp

-from astrai.parallel.signal_handler import install_early_signal_handlers
+from astrai.signal_handler import install_early_signal_handlers

 logger = logging.getLogger(__name__)

@@ -5,7 +5,7 @@ import torch.distributed as dist

 from astrai.config import TrainConfig
 from astrai.parallel.setup import spawn_parallel_fn
-from astrai.parallel.signal_handler import (
+from astrai.signal_handler import (
    register_signal_handlers,
    unregister_signal_handlers,
 )