From 5f1f68499c8508f1695c329ec82c046d23fb8b7f Mon Sep 17 00:00:00 2001 From: Martin Wicke Date: Thu, 15 Aug 2019 13:18:26 -0700 Subject: [PATCH] Replace == with is in test comparing variable identity. PiperOrigin-RevId: 263627688 --- tensorflow/python/eager/wrap_function_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow/python/eager/wrap_function_test.py b/tensorflow/python/eager/wrap_function_test.py index 4b592a5f8df..58cbbbdc82e 100644 --- a/tensorflow/python/eager/wrap_function_test.py +++ b/tensorflow/python/eager/wrap_function_test.py @@ -235,8 +235,8 @@ class WrapFunctionTest(test.TestCase): self.assertIs(g_var_collection[0], v3_holder[0]) # Both have only one value, and their values aren't equal. So no sharing. - self.assertNotEqual(g_wrapped.graph.get_collection(ops.GraphKeys.LOSSES), - f_wrapped.graph.get_collection(ops.GraphKeys.LOSSES)) + self.assertIsNot(g_wrapped.graph.get_collection(ops.GraphKeys.LOSSES[0]), + f_wrapped.graph.get_collection(ops.GraphKeys.LOSSES)[0]) def testGradientsOfPrune(self):