Skip to content

Commit 8cfcbde

Browse files
Update upgrade test to create kueue resources with specfic pre-defined names
1 parent 9295a8e commit 8cfcbde

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

tests/upgrade/raycluster_sdk_upgrade_test.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,10 @@
1010
from codeflare_sdk.utils.kube_api_helpers import _kube_api_error_handling
1111

1212
namespace = "test-ns-rayupgrade"
13+
# Global variables for kueue resources
14+
cluster_queue = "cluster-queue-mnist"
15+
flavor = "default-flavor-mnist"
16+
local_queue = "local-queue-mnist"
1317

1418

1519
# Creates a Ray cluster
@@ -18,7 +22,9 @@ def setup_method(self):
1822
initialize_kubernetes_client(self)
1923
create_namespace_with_name(self, namespace)
2024
try:
21-
create_kueue_resources(self)
25+
create_cluster_queue(self, cluster_queue, flavor)
26+
create_resource_flavor(self, flavor)
27+
create_local_queue(self, cluster_queue, local_queue)
2228
except Exception as e:
2329
delete_namespace(self)
2430
delete_kueue_resources(self)

0 commit comments

Comments
 (0)