Skip to content

Commit 3674cfa

Browse files
Enhance widgets unit tests and increase coverage
1 parent bd60fca commit 3674cfa

File tree

2 files changed

+249
-111
lines changed

2 files changed

+249
-111
lines changed

src/codeflare_sdk/common/widgets/widgets.py

+11-6
Original file line numberDiff line numberDiff line change
@@ -137,6 +137,7 @@ def _on_delete_button_click(self, b):
137137

138138
# Refresh the dataframe
139139
new_df = _fetch_cluster_data(namespace)
140+
self.ray_clusters_df = new_df
140141
if new_df.empty:
141142
self.classification_widget.close()
142143
self.delete_button.close()
@@ -385,15 +386,19 @@ def _fetch_cluster_data(namespace):
385386
namespaces = [item.namespace for item in rayclusters]
386387
num_workers = [item.num_workers for item in rayclusters]
387388
head_extended_resources = [
388-
f"{list(item.head_extended_resources.keys())[0]}: {list(item.head_extended_resources.values())[0]}"
389-
if item.head_extended_resources
390-
else "0"
389+
(
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"
393+
)
391394
for item in rayclusters
392395
]
393396
worker_extended_resources = [
394-
f"{list(item.worker_extended_resources.keys())[0]}: {list(item.worker_extended_resources.values())[0]}"
395-
if item.worker_extended_resources
396-
else "0"
397+
(
398+
f"{list(item.worker_extended_resources.keys())[0]}: {list(item.worker_extended_resources.values())[0]}"
399+
if item.worker_extended_resources
400+
else "0"
401+
)
397402
for item in rayclusters
398403
]
399404
head_cpu_requests = [

0 commit comments

Comments
 (0)