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:
parent
c40d59d7c7
commit
e1b6966d43
@ -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)
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user