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

Fix the KeyError in multi_worker_test_base.py caused by __iter__ returning (key, value) which gets interpreted as the key (and hence no value is found).

Browse Source 
PiperOrigin-RevId: 236163036
This commit is contained in:
Rick Chao 2019-02-28 11:35:32 -08:00 committed by TensorFlower Gardener
parent b9f81dfea1
commit 5139b72eff
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/contrib/distribute/python/multi_worker_test_base.py
View File

@ -347,9 +347,9 @@ class MockOsEnv(collections.Mapping):
def __iter__(self): def __iter__(self):
if not hasattr(self._thread_local, 'dict'): if not hasattr(self._thread_local, 'dict'):
self._thread_local.dict = dict() self._thread_local.dict = dict()
for x in self._thread_local.dict.items(): for x in self._thread_local.dict:
yield x yield x
for x in self._dict.items(): for x in self._dict:
yield x yield x
def __len__(self): def __len__(self):