Skip to content

Commit b61771c

Browse files
committed
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

Lines changed: 7 additions & 1 deletion
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

Lines changed: 2 additions & 0 deletions
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)