From 356de62421989253daed47b37960b3adcf0774cc Mon Sep 17 00:00:00 2001 From: Hye Soo Yang Date: Thu, 12 Dec 2019 15:19:04 -0800 Subject: [PATCH] Fix for b/146078486. In py3, range() returns iterable object. PiperOrigin-RevId: 285282691 Change-Id: I4e6520e5fa6e5df62e88c8fb8a0296ba86fb8465 --- .../distribute/collective_all_reduce_strategy_test.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tensorflow/python/distribute/collective_all_reduce_strategy_test.py b/tensorflow/python/distribute/collective_all_reduce_strategy_test.py index 14ed6b071d1..0c0bff429e6 100644 --- a/tensorflow/python/distribute/collective_all_reduce_strategy_test.py +++ b/tensorflow/python/distribute/collective_all_reduce_strategy_test.py @@ -364,9 +364,9 @@ class CollectiveAllReduceStrategyTestBase( computed_value = sess.run([values.select_replica(r, next_element) for r in range(len(devices))]) if ignore_order: - self.assertCountEqual(expected_value, computed_value) + self.assertCountEqual(list(expected_value), list(computed_value)) else: - self.assertEqual(expected_value, computed_value) + self.assertEqual(list(expected_value), list(computed_value)) with self.assertRaises(errors.OutOfRangeError): next_element = iterator.get_next() @@ -382,9 +382,9 @@ class CollectiveAllReduceStrategyTestBase( computed_value = sess.run([values.select_replica(r, next_element) for r in range(len(devices))]) if ignore_order: - self.assertCountEqual(expected_value, computed_value) + self.assertCountEqual(list(expected_value), list(computed_value)) else: - self.assertEqual(expected_value, computed_value) + self.assertEqual(list(expected_value), list(computed_value)) class DistributedCollectiveAllReduceStrategyTest(