experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

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

Browse Source 
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
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/python/eager/remote_cluster_test.py
View File

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