Update kueue version (#504)

* update kueue version to 0.12.2

Signed-off-by: Piotr Pawłowski <ppawl@google.com>

---------

Signed-off-by: Piotr Pawłowski <ppawl@google.com>

Author: pawloch00

Makefile

@@ -1,7 +1,7 @@
 KUEUE_REPO=https://github.com/kubernetes-sigs/kueue.git
 
 KUBECTL_VERSION := $(shell curl -L -s https://dl.k8s.io/release/stable.txt)
-KUEUE_VERSION=v0.10.0
+KUEUE_VERSION=v0.12.2
 KJOB_VERSION=v0.1.0
 
 OS := $(shell uname -s | tr A-Z a-z)

src/xpk/core/blueprint/blueprint_generator.py

@@ -214,7 +214,7 @@ def generate_a3_mega_blueprint(
         settings={
             "kueue": {
                 "install": True,
-                "version": "v0.10.0",  # TAS feature-gates is enabled in CT
+                "version": "v0.12.2",  # TAS feature-gates is enabled in CT
                 "config_path": f'$(ghpc_stage("{blueprint_name}"))/kueue-xpk-configuration.yaml.tftpl',
                 "config_template_vars": {
                     "num_chips": num_chips,
@@ -574,7 +574,7 @@ def generate_a3_ultra_blueprint(
         settings={
             "kueue": {
                 "install": True,
-                "version": "v0.10.0",  # TAS feature-gates is enabled in CT
+                "version": "v0.12.2",  # TAS feature-gates is enabled in CT
                 "config_path": f'$(ghpc_stage("{blueprint_name}"))/kueue-xpk-configuration.yaml.tftpl',
                 "config_template_vars": {"num_chips": num_chips},
             },
@@ -842,7 +842,7 @@ def generate_a4_blueprint(
         settings={
             "kueue": {
                 "install": True,
-                "version": "v0.10.0",  # TAS feature-gates is enabled in CT
+                "version": "v0.12.2",  # TAS feature-gates is enabled in CT
                 "config_path": f'$(ghpc_stage("{blueprint_name}"))/kueue-xpk-configuration.yaml.tftpl',
                 "config_template_vars": {"num_chips": num_chips},
             },

src/xpk/core/kueue.py

@@ -40,7 +40,7 @@
     SystemCharacteristics,
 )
 
-KUEUE_VERSION = 'v0.10.0'
+KUEUE_VERSION = 'v0.12.2'
 CLUSTER_QUEUE_NAME = 'cluster-queue'
 LOCAL_QUEUE_NAME = 'multislice-queue'
 WAIT_FOR_KUEUE_TIMEOUT = '5m'

src/xpk/core/tests/data/a3_mega.yaml

@@ -98,7 +98,7 @@ deployment_groups:
     settings:
       kueue:
         install: true
-        version: "v0.10.0"
+        version: "v0.12.2"
         config_path: $(ghpc_stage("xpk-gke-a3-megagpu"))/kueue-xpk-configuration.yaml.tftpl
         config_template_vars:
           num_chips: 16

src/xpk/core/tests/data/a3_mega_spot.yaml

@@ -97,7 +97,7 @@ deployment_groups:
     settings:
       kueue:
         install: true
-        version: "v0.10.0"
+        version: "v0.12.2"
         config_path: $(ghpc_stage("xpk-gke-a3-megagpu"))/kueue-xpk-configuration.yaml.tftpl
         config_template_vars:
           num_chips: 16

src/xpk/core/tests/data/a3_ultra.yaml

@@ -153,7 +153,7 @@ deployment_groups:
     settings:
       kueue:
         install: true
-        version: v0.10.0  # TAS feature-gates is enabled in CT
+        version: v0.12.2  # TAS feature-gates is enabled in CT
         config_path: $(ghpc_stage("xpk-gke-a3-ultra"))/kueue-xpk-configuration.yaml.tftpl
         config_template_vars:
          num_chips: 16

src/xpk/core/tests/data/a4.yaml

@@ -165,7 +165,7 @@ deployment_groups:
     settings:
       kueue:
         install: true
-        version: v0.10.0
+        version: v0.12.2
         config_path: $(ghpc_stage("xpk-gke-a4"))/kueue-xpk-configuration.yaml.tftpl
         config_template_vars:
           num_chips: 16