From 6572be4f9876cd854e05fe9ef1e4165eb273d4bd Mon Sep 17 00:00:00 2001 From: ViperEkura <3081035982@qq.com> Date: Fri, 7 Aug 2026 18:30:35 +0800 Subject: [PATCH] fix: prevent signal handler test from racing with training completion - Set n_epoch=99999 so training runs until parent delivers signal instead of finishing too fast on CPU - Drop ready-file deadline from 30s to 10s --- tests/trainer/test_signal_handler.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/trainer/test_signal_handler.py b/tests/trainer/test_signal_handler.py index e95cd28..5ad1b9b 100644 --- a/tests/trainer/test_signal_handler.py +++ b/tests/trainer/test_signal_handler.py @@ -71,7 +71,7 @@ def _inner_run(batch_per_device, ckpt_interval, ckpt_dir, ready_file): optimizer_fn=optimizer_fn, scheduler_fn=scheduler_fn, ckpt_dir=ckpt_dir, - n_epoch=1, + n_epoch=99999, batch_per_device=batch_per_device, ckpt_interval=ckpt_interval, grad_accum_steps=1, @@ -94,7 +94,7 @@ def _spawn_train_and_signal(ckpt_dir, sig, timeout=120): ) p.start() - deadline = time.time() + 30 + deadline = time.time() + 10 while time.time() < deadline: if os.path.exists(ready_file): with open(ready_file) as f: