From 8f55a1c295620c695a01b5f61a92d365a5833de8 Mon Sep 17 00:00:00 2001 From: Milica Makevic Date: Thu, 6 Mar 2025 18:00:33 +0000 Subject: [PATCH] Increase timeout for multi_process_runner_test subtest --- tensorflow/python/distribute/multi_process_runner_test.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tensorflow/python/distribute/multi_process_runner_test.py b/tensorflow/python/distribute/multi_process_runner_test.py index 15e32e78a9cefa..25e9cc280a7403 100644 --- a/tensorflow/python/distribute/multi_process_runner_test.py +++ b/tensorflow/python/distribute/multi_process_runner_test.py @@ -433,7 +433,7 @@ def fn(): return_output=True) mpr.start() with self.assertRaises(ValueError) as cm: - mpr.join(timeout=10) + mpr.join(timeout=20) self.assertGreater( sum(['Running' in msg for msg in cm.exception.mpr_result.stdout]), 1) @@ -479,6 +479,7 @@ def test_auto_restart_terminate(self): # Tasks terminated by the user should also be restarted. def fn(counter): + logging.warning(f"worker fn {counter.value}") counter.value += 1 if counter.value == 1: time.sleep(100) @@ -493,7 +494,8 @@ def fn(counter): args=(counter,), auto_restart=True) mpr.start() - time.sleep(3) + time.sleep(10) + logging.warning("terminating worker!") mpr.terminate('worker', 0) mpr.join(timeout=20) self.assertEqual(counter.value, 2)