File tree 2 files changed +254
-111
lines changed
src/codeflare_sdk/common/widgets
2 files changed +254
-111
lines changed Original file line number Diff line number Diff line change 34
34
get_api_client ,
35
35
)
36
36
37
+
37
38
class RayClusterManagerWidgets :
38
39
def __init__ (self , ray_clusters_df : pd .DataFrame , namespace : str = None ):
39
40
from ...ray .cluster .cluster import get_current_namespace
@@ -137,6 +138,7 @@ def _on_delete_button_click(self, b):
137
138
138
139
# Refresh the dataframe
139
140
new_df = _fetch_cluster_data (namespace )
141
+ self .ray_clusters_df = new_df
140
142
if new_df .empty :
141
143
self .classification_widget .close ()
142
144
self .delete_button .close ()
@@ -387,15 +389,19 @@ def _fetch_cluster_data(namespace):
387
389
namespaces = [item .namespace for item in rayclusters ]
388
390
num_workers = [item .num_workers for item in rayclusters ]
389
391
head_extended_resources = [
390
- f"{ list (item .head_extended_resources .keys ())[0 ]} : { list (item .head_extended_resources .values ())[0 ]} "
391
- if item .head_extended_resources
392
- else "0"
392
+ (
393
+ f"{ list (item .head_extended_resources .keys ())[0 ]} : { list (item .head_extended_resources .values ())[0 ]} "
394
+ if item .head_extended_resources
395
+ else "0"
396
+ )
393
397
for item in rayclusters
394
398
]
395
399
worker_extended_resources = [
396
- f"{ list (item .worker_extended_resources .keys ())[0 ]} : { list (item .worker_extended_resources .values ())[0 ]} "
397
- if item .worker_extended_resources
398
- else "0"
400
+ (
401
+ f"{ list (item .worker_extended_resources .keys ())[0 ]} : { list (item .worker_extended_resources .values ())[0 ]} "
402
+ if item .worker_extended_resources
403
+ else "0"
404
+ )
399
405
for item in rayclusters
400
406
]
401
407
head_cpu_requests = [
You can’t perform that action at this time.
0 commit comments