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):
117
117
zone = None ,
118
118
tpu = ['test-tpu-1' ],
119
119
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' )
121
122
122
123
actual_cluster_spec = tpu_cluster_resolver .cluster_spec ()
123
124
expected_proto = """
@@ -170,6 +171,7 @@ def testSimpleSuccessfulRetrieval(self):
170
171
project = 'test-project' ,
171
172
zone = 'us-central1-c' ,
172
173
tpu = ['test-tpu-1' ],
174
+ coordinator_name = 'coordinator' ,
173
175
coordinator_address = '10.128.1.5:10203' ,
174
176
credentials = None ,
175
177
service = self .mock_service_client (tpu_map = tpu_map ))
@@ -196,6 +198,7 @@ def testNewNetworkEndpointFormat(self):
196
198
project = 'test-project' ,
197
199
zone = 'us-central1-c' ,
198
200
tpu = 'test-tpu-1' ,
201
+ coordinator_name = 'coordinator' ,
199
202
coordinator_address = '10.128.1.5:10203' ,
200
203
credentials = None ,
201
204
service = self .mock_service_client (tpu_map = tpu_map ))
@@ -239,7 +242,8 @@ def testPodResolution(self):
239
242
tpu_cluster_resolver = TPUClusterResolver (
240
243
tpu = 'test-tpu-1' ,
241
244
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' )
243
247
244
248
actual_cluster_spec = tpu_cluster_resolver .cluster_spec ()
245
249
expected_proto = """
You can’t perform that action at this time.
0 commit comments