File tree Expand file tree Collapse file tree 1 file changed +6
-2
lines changed
tensorflow/contrib/cluster_resolver/python/training Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Original file line number Diff line number Diff line change @@ -117,7 +117,8 @@ def testRetrieveProjectAndZoneFromMetadata(self):
117117 zone = None ,
118118 tpu = ['test-tpu-1' ],
119119 credentials = None ,
120- service = self .mock_service_client (tpu_map = tpu_map ))
120+ service = self .mock_service_client (tpu_map = tpu_map ),
121+ coordinator_name = 'coordinator' )
121122
122123 actual_cluster_spec = tpu_cluster_resolver .cluster_spec ()
123124 expected_proto = """
@@ -170,6 +171,7 @@ def testSimpleSuccessfulRetrieval(self):
170171 project = 'test-project' ,
171172 zone = 'us-central1-c' ,
172173 tpu = ['test-tpu-1' ],
174+ coordinator_name = 'coordinator' ,
173175 coordinator_address = '10.128.1.5:10203' ,
174176 credentials = None ,
175177 service = self .mock_service_client (tpu_map = tpu_map ))
@@ -196,6 +198,7 @@ def testNewNetworkEndpointFormat(self):
196198 project = 'test-project' ,
197199 zone = 'us-central1-c' ,
198200 tpu = 'test-tpu-1' ,
201+ coordinator_name = 'coordinator' ,
199202 coordinator_address = '10.128.1.5:10203' ,
200203 credentials = None ,
201204 service = self .mock_service_client (tpu_map = tpu_map ))
@@ -239,7 +242,8 @@ def testPodResolution(self):
239242 tpu_cluster_resolver = TPUClusterResolver (
240243 tpu = 'test-tpu-1' ,
241244 credentials = None ,
242- service = self .mock_service_client (tpu_map = tpu_map ))
245+ service = self .mock_service_client (tpu_map = tpu_map ),
246+ coordinator_name = 'coordinator' )
243247
244248 actual_cluster_spec = tpu_cluster_resolver .cluster_spec ()
245249 expected_proto = """
You can’t perform that action at this time.
0 commit comments