Skip to content

Commit 693a019

Browse files
Enhance widgets unit tests and increase coverage
1 parent 3385e34 commit 693a019

File tree

2 files changed

+254
-111
lines changed

2 files changed

+254
-111
lines changed

src/codeflare_sdk/common/widgets/widgets.py

+12-6
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@
3434
get_api_client,
3535
)
3636

37+
3738
class 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 = [

0 commit comments

Comments
 (0)