@@ -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 = """
0 commit comments