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

Fix for b/146078486. In py3, range() returns iterable object.

Browse Source 
PiperOrigin-RevId: 285282691
Change-Id: I4e6520e5fa6e5df62e88c8fb8a0296ba86fb8465
This commit is contained in:
Hye Soo Yang 2019-12-12 15:19:04 -08:00 committed by TensorFlower Gardener
parent 9c4b6d9df8
commit 356de62421
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/python/distribute/collective_all_reduce_strategy_test.py
View File

@ -364,9 +364,9 @@ class CollectiveAllReduceStrategyTestBase(
computed_value = sess.run([values.select_replica(r, next_element) computed_value = sess.run([values.select_replica(r, next_element)
for r in range(len(devices))]) for r in range(len(devices))])
if ignore_order: if ignore_order:
self.assertCountEqual(expected_value, computed_value) self.assertCountEqual(list(expected_value), list(computed_value))
else: else:
self.assertEqual(expected_value, computed_value) self.assertEqual(list(expected_value), list(computed_value))
with self.assertRaises(errors.OutOfRangeError): with self.assertRaises(errors.OutOfRangeError):
next_element = iterator.get_next() next_element = iterator.get_next()
@ -382,9 +382,9 @@ class CollectiveAllReduceStrategyTestBase(
computed_value = sess.run([values.select_replica(r, next_element) computed_value = sess.run([values.select_replica(r, next_element)
for r in range(len(devices))]) for r in range(len(devices))])
if ignore_order: if ignore_order:
self.assertCountEqual(expected_value, computed_value) self.assertCountEqual(list(expected_value), list(computed_value))
else: else:
self.assertEqual(expected_value, computed_value) self.assertEqual(list(expected_value), list(computed_value))
class DistributedCollectiveAllReduceStrategyTest( class DistributedCollectiveAllReduceStrategyTest(