Skip to content
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.

Commit b61771c

Browse files
committedApr 2, 2024·
Update tests, Rebase
1 parent 27e023c commit b61771c

File tree

2 files changed

+9
-1
lines changed

2 files changed

+9
-1
lines changed
 

‎src/codeflare_sdk/cluster/__init__.py

+7-1
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,12 @@
1313
AppWrapper,
1414
)
1515

16-
from .cluster import Cluster, ClusterConfiguration, get_cluster, list_all_queued, list_all_clusters
16+
from .cluster import (
17+
Cluster,
18+
ClusterConfiguration,
19+
get_cluster,
20+
list_all_queued,
21+
list_all_clusters,
22+
)
1723

1824
from .awload import AWManager

‎tests/unit_test.py

+2
Original file line numberDiff line numberDiff line change
@@ -2462,6 +2462,8 @@ def custom_side_effect(group, version, namespace, plural, **kwargs):
24622462
return get_ray_obj("ray.io", "v1", "ns", "rayclusters")
24632463
elif plural == "appwrappers":
24642464
return get_aw_obj("workload.codeflare.dev", "v1beta1", "ns", "appwrappers")
2465+
elif plural == "localqueues":
2466+
return get_local_queue("kueue.x-k8s.io", "v1beta1", "ns", "localqueues")
24652467

24662468
mocker.patch(
24672469
"kubernetes.client.CustomObjectsApi.list_namespaced_custom_object", get_aw_obj

0 commit comments

Comments
 (0)
Please sign in to comment.