Fix broken test (invalid string comparison in py3)
authorChristopher Suter <cgs@google.com>
Tue, 6 Mar 2018 19:21:20 +0000 (11:21 -0800)
committerTensorFlower Gardener <gardener@tensorflow.org>
Tue, 6 Mar 2018 19:25:52 +0000 (11:25 -0800)
PiperOrigin-RevId: 188051422

tensorflow/contrib/cluster_resolver/python/training/tpu_cluster_resolver_test.py

index b7d56fc..48c3f6b 100644 (file)
@@ -362,10 +362,10 @@ class TPUClusterResolverTest(test.TestCase):
     self.assertTrue(tpu_cluster_resolver._inGke())
     self.assertEqual(
         compat.as_bytes('grpc://10.120.27.5:8470'),
-        tpu_cluster_resolver._gkeMaster())
+        compat.as_bytes(tpu_cluster_resolver._gkeMaster()))
     self.assertEqual(
         compat.as_bytes('grpc://10.120.27.5:8470'),
-        tpu_cluster_resolver.get_master())
+        compat.as_bytes(tpu_cluster_resolver.get_master()))
     del os.environ['KUBE_GOOGLE_CLOUD_TPU_ENDPOINTS']