Skip to content

Commit f1c94f0

Browse files
Enhance widgets unit tests and increase coverage
1 parent 9483f60 commit f1c94f0

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()
@@ -387,15 +388,19 @@ def _fetch_cluster_data(namespace):
387388
namespaces = [item.namespace for item in rayclusters]
388389
num_workers = [item.num_workers for item in rayclusters]
389390
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"
391+
(
392+
f"{list(item.head_extended_resources.keys())[0]}: {list(item.head_extended_resources.values())[0]}"
393+
if item.head_extended_resources
394+
else "0"
395+
)
393396
for item in rayclusters
394397
]
395398
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"
399+
(
400+
f"{list(item.worker_extended_resources.keys())[0]}: {list(item.worker_extended_resources.values())[0]}"
401+
if item.worker_extended_resources
402+
else "0"
403+
)
399404
for item in rayclusters
400405
]
401406
head_cpu_requests = [

0 commit comments

Comments
 (0)