@@ -221,9 +221,9 @@ def get_cluster_key_and_resource_group(self, worker_id: str):
221
221
return cluster_key , resource_group
222
222
223
223
def _create_cluster_report (
224
- self , worker_id = None , cluster_key = None , max_gpus = None , max_workers = None , resource_groups = None
224
+ self , worker_id = None , cluster_key = None , max_gpus = None , max_workers = None , resource_groups = None , pending = False
225
225
):
226
- # type: (Optional[str], Optional[str], Optional[int], Optional[int], Optional[List[str]]) -> ()
226
+ # type: (Optional[str], Optional[str], Optional[int], Optional[int], Optional[List[str]], bool ) -> ()
227
227
"""
228
228
Set up a cluster report for the enterprise server dashboard feature.
229
229
If a worker_id is provided, cluster_key and resource_groups are inferred from it.
@@ -247,7 +247,8 @@ def _create_cluster_report(
247
247
cluster_key = cluster_key ,
248
248
max_gpus = max_gpus ,
249
249
max_workers = max_workers ,
250
- resource_groups = resource_groups
250
+ resource_groups = resource_groups ,
251
+ pending = pending ,
251
252
)
252
253
except Exception as ex :
253
254
print ("Error: failed setting cluster report: {}" .format (ex ))
@@ -278,9 +279,9 @@ def setup_daemon_cluster_report(self, worker_id=None, cluster_key=None, max_work
278
279
worker_id = worker_id ,
279
280
cluster_key = cluster_key ,
280
281
max_workers = max_workers ,
281
- resource_groups = resource_groups
282
+ resource_groups = resource_groups ,
283
+ pending = True ,
282
284
)
283
- self ._cluster_report .pending = True
284
285
285
286
def _daemon (self ):
286
287
last_cluster_report = 0
0 commit comments