From 053f39e766b6e3bf8b6bcf1c932857fbccd41537 Mon Sep 17 00:00:00 2001 From: Martin Wicke Date: Thu, 15 Aug 2019 13:19:13 -0700 Subject: [PATCH] Use id() for Tensor identity comparison. PiperOrigin-RevId: 263627848 --- tensorflow/python/training/tracking/util_test.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tensorflow/python/training/tracking/util_test.py b/tensorflow/python/training/tracking/util_test.py index 5045f5af5c3..9fec0841b1a 100644 --- a/tensorflow/python/training/tracking/util_test.py +++ b/tensorflow/python/training/tracking/util_test.py @@ -1413,8 +1413,9 @@ class TemplateTests(parameterized.TestCase, test.TestCase): v1_save, _, v2_save, manual_scope, manual_scope_v = save_template() six.assertCountEqual( self, - [v1_save, v2_save, manual_scope, manual_scope_v, save_template], - trackable_utils.list_objects(save_template)) + [id(v1_save), id(v2_save), id(manual_scope), + id(manual_scope_v), id(save_template)], + map(id, trackable_utils.list_objects(save_template))) manual_dep, = manual_scope._checkpoint_dependencies self.assertEqual("in_manual_scope", manual_dep.name) self.assertIs(manual_scope_v, manual_dep.ref)