@@ -618,39 +618,39 @@ def _fetch_cluster_data(namespace):
618618 namespaces = [item .namespace for item in rayclusters ]
619619 head_extended_resources = [
620620 f"{ list (item .head_extended_resources .keys ())[0 ]} : { list (item .head_extended_resources .values ())[0 ]} "
621- if item .head_extended_resources else "nvidia.com/gpu: 0"
621+ if item .head_extended_resources else "0"
622622 for item in rayclusters
623623 ]
624624 worker_extended_resources = [
625625 f"{ list (item .worker_extended_resources .keys ())[0 ]} : { list (item .worker_extended_resources .values ())[0 ]} "
626- if item .worker_extended_resources else "nvidia.com/gpu: 0"
626+ if item .worker_extended_resources else "0"
627627 for item in rayclusters
628628 ]
629629 head_cpu_requests = [item .head_cpu_requests if item .head_cpu_requests else 0 for item in rayclusters ]
630630 head_cpu_limits = [item .head_cpu_limits if item .head_cpu_limits else 0 for item in rayclusters ]
631+ head_cpu_rl = [f"{ requests } ~{ limits } " for requests , limits in zip (head_cpu_requests , head_cpu_limits )]
631632 head_mem_requests = [item .head_mem_requests if item .head_mem_requests else 0 for item in rayclusters ]
632633 head_mem_limits = [item .head_mem_limits if item .head_mem_limits else 0 for item in rayclusters ]
634+ head_mem_rl = [f"{ requests } ~{ limits } " for requests , limits in zip (head_mem_requests , head_mem_limits )]
633635 worker_cpu_requests = [item .worker_cpu_requests if item .worker_cpu_requests else 0 for item in rayclusters ]
634636 worker_cpu_limits = [item .worker_cpu_limits if item .worker_cpu_limits else 0 for item in rayclusters ]
637+ worker_cpu_rl = [f"{ requests } ~{ limits } " for requests , limits in zip (worker_cpu_requests , worker_cpu_limits )]
635638 worker_mem_requests = [item .worker_mem_requests if item .worker_mem_requests else 0 for item in rayclusters ]
636639 worker_mem_limits = [item .worker_mem_limits if item .worker_mem_limits else 0 for item in rayclusters ]
640+ worker_mem_rl = [f"{ requests } ~{ limits } " for requests , limits in zip (worker_mem_requests , worker_mem_limits )]
637641 status = [item .status .name for item in rayclusters ]
638642
639643 status = [format_status (item .status ) for item in rayclusters ]
640644
641645 data = {
642- "name" : names ,
643- "namespace" : namespaces ,
644- "head gpus" : head_extended_resources ,
645- "worker gpus" : worker_extended_resources ,
646- "head cpu requests" : head_cpu_requests ,
647- "head cpu limits" : head_cpu_limits ,
648- "head memory requests" : head_mem_requests ,
649- "head memory limits" : head_mem_limits ,
650- "worker cpu requests" : worker_cpu_requests ,
651- "worker cpu limits" : worker_cpu_limits ,
652- "worker memory requests" : worker_mem_requests ,
653- "worker memory limits" : worker_mem_limits ,
646+ "Name" : names ,
647+ "Namespace" : namespaces ,
648+ "Head GPUs" : head_extended_resources ,
649+ "Worker GPUs" : worker_extended_resources ,
650+ "Head CPU Req~Lim" : head_cpu_rl ,
651+ "Head Memory Req~Lim" : head_mem_rl ,
652+ "Worker CPU Req~Lim" : worker_cpu_rl ,
653+ "Worker Memory Req~Lim" : worker_mem_rl ,
654654 "status" : status
655655 }
656656 return pd .DataFrame (data )
@@ -660,39 +660,39 @@ def list_cluster_details(namespace: str):
660660 df = _fetch_cluster_data (namespace )
661661
662662 my_output = widgets .Output ()
663- if df ["name " ].empty :
663+ if df ["Name " ].empty :
664664 print (f"No clusters found in the { namespace } namespace." )
665665 else :
666666 classification_widget = widgets .ToggleButtons (
667- options = df ["name " ].tolist (), value = None ,
667+ options = df ["Name " ].tolist (), value = None ,
668668 description = 'Select an existing cluster:' ,
669669 )
670670
671671 def on_cluster_click (change ):
672672 new_value = change ["new" ]
673673 my_output .clear_output ()
674674 with my_output :
675- display (HTML (df [df ["name " ]== new_value ][["name " , "namespace " , "head gpus " , "worker gpus" , "head cpu requests" , "head cpu limits " , "head memory requests " , "head memory limits " , "worker memory requests" , "worker memory limits " , "status" ]].to_html (escape = False , index = False , border = 2 )))
675+ display (HTML (df [df ["Name " ]== new_value ][["Name " , "Namespace " , "Head GPUs " , "Head CPU Req~Lim" , "Head Memory Req~Lim " , "Worker GPUs " , "Worker CPU Req~Lim " , "Worker Memory Req~Lim " , "status" ]].to_html (escape = False , index = False , border = 2 )))
676676
677677 classification_widget .observe (on_cluster_click , names = "value" )
678678 display (widgets .VBox ([classification_widget , my_output ]))
679679
680680 def on_delete_button_clicked (b ):
681681 cluster_name = classification_widget .value
682- namespace = df [df ["name " ]== classification_widget .value ]["namespace " ].values [0 ]
682+ namespace = df [df ["Name " ]== classification_widget .value ]["Namespace " ].values [0 ]
683683 delete_cluster (cluster_name , namespace )
684684 my_output .clear_output ()
685685 print (f"Cluster { cluster_name } in the { namespace } namespace was deleted successfully." )
686686 # Refresh the dataframe
687687 new_df = _fetch_cluster_data (namespace )
688- classification_widget .options = new_df ["name " ].tolist ()
688+ classification_widget .options = new_df ["Name " ].tolist ()
689689
690690
691691 # out Output widget is used to execute JavaScript code to open the Ray dashboard URL in a new browser tab
692692 out = widgets .Output ()
693693 def on_ray_dashboard_button_clicked (b ):
694694 cluster_name = classification_widget .value
695- namespace = df [df ["name " ]== classification_widget .value ]["namespace " ].values [0 ]
695+ namespace = df [df ["Name " ]== classification_widget .value ]["Namespace " ].values [0 ]
696696
697697 cluster = Cluster (ClusterConfiguration (cluster_name , namespace ))
698698 dashboard_url = cluster .cluster_dashboard_uri ()
@@ -704,7 +704,7 @@ def on_ray_dashboard_button_clicked(b):
704704
705705 def on_list_jobs_button_clicked (b ):
706706 cluster_name = classification_widget .value
707- namespace = df [df ["name " ]== classification_widget .value ]["namespace " ].values [0 ]
707+ namespace = df [df ["Name " ]== classification_widget .value ]["Namespace " ].values [0 ]
708708
709709 cluster = Cluster (ClusterConfiguration (cluster_name , namespace ))
710710 dashboard_url = cluster .cluster_dashboard_uri ()
0 commit comments