Skip to content

Commit

Permalink
Update upgrade test to create kueue resources with specfic pre-define…
Browse files Browse the repository at this point in the history
…d names
  • Loading branch information
abhijeet-dhumal authored and openshift-merge-bot[bot] committed Jun 7, 2024
1 parent 9295a8e commit 8cfcbde
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion tests/upgrade/raycluster_sdk_upgrade_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,10 @@
from codeflare_sdk.utils.kube_api_helpers import _kube_api_error_handling

namespace = "test-ns-rayupgrade"
# Global variables for kueue resources
cluster_queue = "cluster-queue-mnist"
flavor = "default-flavor-mnist"
local_queue = "local-queue-mnist"


# Creates a Ray cluster
Expand All @@ -18,7 +22,9 @@ def setup_method(self):
initialize_kubernetes_client(self)
create_namespace_with_name(self, namespace)
try:
create_kueue_resources(self)
create_cluster_queue(self, cluster_queue, flavor)
create_resource_flavor(self, flavor)
create_local_queue(self, cluster_queue, local_queue)
except Exception as e:
delete_namespace(self)
delete_kueue_resources(self)
Expand Down

0 comments on commit 8cfcbde

Please sign in to comment.