bug fix in trainer_test

chenyushuo · chenyushuo · commit 7d8e99aa4eef · 2025-09-26T14:39:06.000+08:00
diff --git a/tests/trainer/trainer_test.py b/tests/trainer/trainer_test.py
@@ -780,6 +780,7 @@ def test_trainer(self):
                 print(f"Checkpoint check at {state_dict_iteration} iteration passed.")
 
             time.sleep(1)
+        trainer_process.join()
 
     def tearDown(self):
         # remove dir only when the test passed
