We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9295a8e commit 8cfcbdeCopy full SHA for 8cfcbde
tests/upgrade/raycluster_sdk_upgrade_test.py
@@ -10,6 +10,10 @@
10
from codeflare_sdk.utils.kube_api_helpers import _kube_api_error_handling
11
12
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"
17
18
19
# Creates a Ray cluster
@@ -18,7 +22,9 @@ def setup_method(self):
22
initialize_kubernetes_client(self)
23
create_namespace_with_name(self, namespace)
20
24
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)
28
except Exception as e:
29
delete_namespace(self)
30
delete_kueue_resources(self)
0 commit comments