Use coordinator.join instead of joining threads to make sure thread exceptions are reliably thrown.

PiperOrigin-RevId: 314842568
Change-Id: I8be3f86eabeb4f44dd96b2303da720761bb1ea2a
This commit is contained in:
Haoyu Zhang 2020-06-04 18:17:24 -07:00 committed by TensorFlower Gardener
parent c40d59d7c7
commit e1b6966d43

View File

@ -530,8 +530,7 @@ class DynamicClusterTest(test.TestCase, parameterized.TestCase):
threads.append(threading.Thread(target=update_server_def_fn))
for t in threads:
t.start()
for t in threads:
t.join()
self._coord.join(threads)
for result in results:
np.testing.assert_array_equal([[2, 2], [2, 2]], result)