File tree Expand file tree Collapse file tree 2 files changed +254
-111
lines changed
src/codeflare_sdk/common/widgets Expand file tree Collapse file tree 2 files changed +254
-111
lines changed Original file line number Diff line number Diff line change 3434 get_api_client ,
3535)
3636
37+
3738class RayClusterManagerWidgets :
3839 def __init__ (self , ray_clusters_df : pd .DataFrame , namespace : str = None ):
3940 from ...ray .cluster .cluster import get_current_namespace
@@ -137,6 +138,7 @@ def _on_delete_button_click(self, b):
137138
138139 # Refresh the dataframe
139140 new_df = _fetch_cluster_data (namespace )
141+ self .ray_clusters_df = new_df
140142 if new_df .empty :
141143 self .classification_widget .close ()
142144 self .delete_button .close ()
@@ -387,15 +389,19 @@ def _fetch_cluster_data(namespace):
387389 namespaces = [item .namespace for item in rayclusters ]
388390 num_workers = [item .num_workers for item in rayclusters ]
389391 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+ )
393397 for item in rayclusters
394398 ]
395399 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+ )
399405 for item in rayclusters
400406 ]
401407 head_cpu_requests = [
You can’t perform that action at this time.
0 commit comments