xpk 0.16.1__tar.gz → 0.17.1__tar.gz

{xpk-0.16.1 → xpk-0.17.1}/.github/release.yaml

@@ -17,9 +17,11 @@
 
 changelog:
   categories:
+  - title: Breaking Changes
+    labels: [release-breaking]
   - title: New Features
     labels: [release-features]
-  - title: Improvments
-    labels: [release-improvments]
+  - title: Improvements
+    labels: [release-improvements]
   - title: Bug fixes
     labels: [release-bugfix]

{xpk-0.16.1 → xpk-0.17.1}/.github/workflows/build_wheels.yaml

@@ -25,9 +25,29 @@ on:
   push:
     tags:
       - "v[0-9]+.[0-9]+.[0-9]+"
+  workflow_dispatch:
 
 jobs:
+  validate_tag:
+    name: Validate Tag Pattern
+    runs-on: ubuntu-latest
+    steps:
+      - name: Check ref
+        run: |
+          # When workflow is dispatched through workflow_dispatch, it could be launched on branches instead of tags.
+          # We want to make sure it is launched with the tag that additionally is a valid version tag.
+          if [[ "${{ github.ref_type }}" != "tag" ]]; then
+            echo "::error::This workflow must be run on a Tag, not a Branch."
+            exit 1
+          fi
+
+          if [[ ! "${{ github.ref_name }}" =~ ^v[0-9]+\.[0-9]+\.[0-9]+$ ]]; then
+            echo "::error::Tag '${{ github.ref_name }}' does not match the required pattern v[0-9]+.[0-9]+.[0-9]+"
+            exit 1
+          fi
+
   build_wheel:
+    needs: [validate_tag]
     uses: ./.github/workflows/reusable_build_wheel.yaml
   approval:
     name: Wait for approval

{xpk-0.16.1 → xpk-0.17.1}/.github/workflows/integration_basic_cluster_create.yaml

@@ -43,7 +43,7 @@ jobs:
           - name: Check xpk installation
             run: xpk version
           - name: Create an XPK Cluster with zero node pools
-            run: xpk cluster create --cluster $EMPTY_CLUSTER_NAME --tpu-type=v4-8  --num-slices=0 --zone=us-central2-b --default-pool-cpu-machine-type=n1-standard-16 --reservation='${{ secrets.GCP_TPU_V4_RESERVATION }}' --custom-cluster-arguments='${{ secrets.CLUSTER_ARGUMENTS }}'
+            run: xpk cluster create --cluster $EMPTY_CLUSTER_NAME --device-type=n2-standard-64-1 --num-slices=0 --zone=us-central2-b --default-pool-cpu-machine-type=n1-standard-16 --on-demand --custom-cluster-arguments='${{ secrets.CLUSTER_ARGUMENTS }}'
           - name: Delete the cluster created
             run: xpk cluster delete --cluster $EMPTY_CLUSTER_NAME --zone=us-central2-b --force
             if: always()
@@ -71,7 +71,7 @@ jobs:
           - name: Check xpk installation
             run: xpk version
           - name: Create a Private XPK Cluster with zero node pools
-            run: xpk cluster create --cluster $PRIVATE_CLUSTER_NAME --private --tpu-type=v4-8  --num-slices=0 --zone=us-central2-b --default-pool-cpu-machine-type=n1-standard-16 --reservation='${{ secrets.GCP_TPU_V4_RESERVATION }}' --custom-cluster-arguments='${CLUSTER_NETWORK_ARGUMENTS}'
+            run: xpk cluster create --cluster $PRIVATE_CLUSTER_NAME --private --device-type=n2-standard-64-1  --num-slices=0 --zone=us-central2-b --default-pool-cpu-machine-type=n1-standard-16 --on-demand --custom-cluster-arguments='${CLUSTER_NETWORK_ARGUMENTS}'
           - name: Verify the created cluster is private
             run: gcloud container clusters describe $PRIVATE_CLUSTER_NAME --region=us-central2 --format="value(privateClusterConfig.enablePrivateNodes)" | grep 'True' || (echo 'The created cluster is not private.' && exit 1)
           - name: Delete the cluster created

{xpk-0.16.1 → xpk-0.17.1}/.github/workflows/integration_storage_tests.yaml

@@ -40,7 +40,7 @@ jobs:
     - name: Check xpk installation
       run: xpk version
     - name: Create cluster
-      run: xpk cluster create --cluster xpk-storage-tests --tpu-type=v4-8  --num-slices=1 --zone=us-central2-b --default-pool-cpu-machine-type=n1-standard-16 --default-pool-cpu-num-nodes=2 --reservation='${{ secrets.GCP_TPU_V4_RESERVATION }}' --enable-gcpfilestore-csi-driver --enable-gcsfuse-csi-driver --custom-cluster-arguments="${CLUSTER_ARGUMENTS}"
+      run: xpk cluster create --cluster xpk-storage-tests --device-type=n2-standard-64-1  --num-slices=1 --zone=us-central2-b --default-pool-cpu-machine-type=n1-standard-16 --default-pool-cpu-num-nodes=2 --on-demand --enable-gcpfilestore-csi-driver --enable-gcsfuse-csi-driver --custom-cluster-arguments="${CLUSTER_ARGUMENTS}"
 
   delete-cluster:
     runs-on: [ubuntu-22.04]
@@ -71,7 +71,7 @@ jobs:
       cancel-in-progress: true
     with:
       cluster-name: xpk-storage-tests
-      tpu-type: v4-8
+      device-type: n2-standard-64-1
       zone: us-central2-b
       storage-type: 'gcsfuse'
       storage-command: 'attach'
@@ -103,7 +103,7 @@ jobs:
       cancel-in-progress: true
     with:
       cluster-name: xpk-storage-tests
-      tpu-type: v4-8
+      device-type: n2-standard-64-1
       zone: us-central2-b
       storage-type: 'gcpfilestore'
       storage-command: 'create'
@@ -134,7 +134,7 @@ jobs:
       cancel-in-progress: true
     with:
       cluster-name: xpk-storage-tests
-      tpu-type: v4-8
+      device-type: n2-standard-64-1
       zone: us-central2-b
       storage-type: 'gcpfilestore'
       storage-command: 'attach'

xpk-0.17.1/.github/workflows/periodic_release.yaml

@@ -0,0 +1,133 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License
+
+name: Periodic Release
+
+on:
+  schedule:
+    - cron: "30 10 * * 1" # At 10:30 on Monday.
+  workflow_dispatch:
+
+permissions:
+  contents: write
+  actions: write
+
+jobs:
+  validate-ref:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Check ref
+        run: |
+          if [[ "${{ github.ref_type }}" != "branch" ]]; then
+            echo "::error::This workflow must be run on a Branch, not a Tag."
+            exit 1
+          fi
+
+          if [[ "${{ github.ref_name }}" != "main" ]]; then
+            echo "::error::This workflow must be run on a main branch."
+            exit 1
+          fi
+
+  compute-release-branch:
+    needs: [validate-ref]
+    runs-on: ubuntu-latest
+    outputs:
+      skip_push: ${{ steps.compute_branch_name.outputs.skip_push }}
+      new_branch_name: ${{ steps.compute_branch_name.outputs.new_branch_name }}
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v4
+        with:
+          fetch-depth: 0
+          fetch-tags: true
+      - name: Compute Next Version
+        id: compute_branch_name
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+        run: |
+          LATEST_BRANCH=$(git branch -r | grep -oE 'origin/release-[0-9]+\.[0-9]+' | sort -V | tail -n 1)
+
+          if [ -z "$LATEST_BRANCH" ]; then
+            echo "No existing release branches found. Cannot determine next version automatically."
+            exit 1
+          fi
+
+          echo "Latest release branch found: $LATEST_BRANCH"
+
+          COMMIT_COUNT=$(git rev-list --count "$LATEST_BRANCH"..HEAD)
+
+          echo "Commits since last release: $COMMIT_COUNT"
+
+          if [ "$COMMIT_COUNT" -eq 0 ]; then
+            echo "No new changes detected compared to $LATEST_BRANCH."
+            echo "Skipping release branch creation."
+            echo "skip_push=true" >> $GITHUB_OUTPUT
+            exit 0
+          fi
+
+          VERSION_STR=$(echo "$LATEST_BRANCH" | sed 's/origin\/release-//')
+          CURRENT_MAJOR=$(echo "$VERSION_STR" | cut -d. -f1)
+          CURRENT_MINOR=$(echo "$VERSION_STR" | cut -d. -f2)
+
+          echo "Retrieving pull request labels between $LATEST_BRANCH and HEAD..."
+          LABELS=$(git log $LATEST_BRANCH..HEAD --pretty=format:"%s" \
+            | grep -oE "#[0-9]+" \
+            | tr -d "#" \
+            | xargs -I {} gh pr view {} --json labels --jq '.labels[].name' \
+            | sort \
+            | uniq)
+
+          echo "Pull request labels:"
+          echo "$LABELS"
+
+          if echo "$LABELS" | grep -q "release-breaking"; then
+            NEW_MAJOR=$((CURRENT_MAJOR + 1))
+            NEW_MINOR=0
+            echo "BREAKING detected. Bumping Major version."
+          else
+            NEW_MAJOR=$CURRENT_MAJOR
+            NEW_MINOR=$((CURRENT_MINOR + 1))
+            echo "No breaking changes detected. Bumping Minor version."
+          fi
+
+          NEW_BRANCH_NAME="release-$NEW_MAJOR.$NEW_MINOR"
+          echo "New release branch calculated: $NEW_BRANCH_NAME"
+          echo "new_branch_name=$NEW_BRANCH_NAME" >> $GITHUB_OUTPUT
+          echo "skip_push=false" >> $GITHUB_OUTPUT
+
+  create-release-branch:
+    needs: [compute-release-branch]
+    environment:
+      name: release
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout repository
+        if: needs.compute-release-branch.outputs.skip_push != 'true'
+        uses: actions/checkout@v4
+        with:
+          fetch-depth: 0
+          fetch-tags: true
+      - name: Push the branch
+        if: needs.compute-release-branch.outputs.skip_push != 'true'
+        run: |
+          NEW_BRANCH_NAME="${{needs.compute-release-branch.outputs.new_branch_name}}"
+          git checkout -b "$NEW_BRANCH_NAME"
+          git push origin "$NEW_BRANCH_NAME"
+
+          echo "Successfully pushed $NEW_BRANCH_NAME"
+      - name: Run release branch versioning
+        if: needs.compute-release-branch.outputs.skip_push != 'true'
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+        run: gh workflow run release_branch_versioning.yaml --ref ${{needs.compute-release-branch.outputs.new_branch_name}}

xpk-0.17.1/.github/workflows/release_branch_versioning.yaml

@@ -0,0 +1,103 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License
+
+name: Release branch versioning
+
+on:
+    push:
+        branches:
+            - "release-*"
+    workflow_dispatch:
+
+permissions:
+    contents: write
+    actions: write
+
+jobs:
+    validate_branch:
+        name: Validate Branch Pattern
+        runs-on: ubuntu-latest
+        steps:
+            - name: Check ref
+              run: |
+                # When workflow is dispatched through workflow_dispatch, it could be launched on any branch or tag.
+                # We want to make sure it is launched with the branch that matches the syntax.
+                if [[ "${{ github.ref_type }}" != "branch" ]]; then
+                  echo "::error::This workflow must be run on a Branch, not a Tag."
+                  exit 1
+                fi
+
+                if [[ ! "${{ github.ref_name }}" =~ ^release- ]]; then
+                  echo "::error::Invalid Branch: Manual runs are only allowed on branches matching 'release-*'. Current branch: ${{ github.ref_name }}"
+                  exit 1
+                fi
+
+    bump_version:
+        needs: [validate_branch]
+        runs-on: ubuntu-latest
+        outputs:
+          tag: ${{ steps.bump.outputs.tag }}
+        steps:
+            - name: Checkout code
+              uses: actions/checkout@v4
+              with:
+                  token: ${{ secrets.GITHUB_TOKEN }}
+                  fetch-depth: 0
+
+            - name: Calculate Next Version
+              id: bump
+              run: |
+                  EXISTING_TAG=$(git tag --points-at HEAD | grep -E '^v[0-9]+\.[0-9]+\.[0-9]+$' | head -n 1)
+
+                  if [ -n "$EXISTING_TAG" ]; then
+                    echo "Commit is already tagged with: $EXISTING_TAG"
+                    exit 1
+                  fi
+
+                  BRANCH_NAME="${{ github.ref_name }}"
+                  BASE_VERSION=${BRANCH_NAME#release-}
+
+                  echo "Base Version derived from branch: $BASE_VERSION"
+
+                  LATEST_TAG=$(git tag -l "v$BASE_VERSION.*" | sort -V | tail -n1)
+
+                  if [ -z "$LATEST_TAG" ]; then
+                    echo "No tags found for this release branch. Starting at patch .0"
+                    NEW_TAG="v$BASE_VERSION.0"
+                  else
+                    echo "Found existing latest tag: $LATEST_TAG"
+                    
+                    VERSION=${LATEST_TAG#v}
+                    
+                    IFS='.' read -r -a parts <<< "$VERSION"
+                    patch=${parts[2]}
+                    
+                    new_patch=$((patch + 1))
+                    NEW_TAG="v$BASE_VERSION.$new_patch"
+                  fi
+
+                  echo "Next Tag: $NEW_TAG"
+                  echo "tag=$NEW_TAG" >> $GITHUB_OUTPUT
+
+            - name: Push Tag
+              run: |
+                  NEW_TAG=${{ steps.bump.outputs.tag }}
+
+                  echo "Pushing tag $NEW_TAG..."
+                  git tag "$NEW_TAG"
+                  git push origin "$NEW_TAG"
+            - name: Run build
+              env:
+                GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+              run: gh workflow run build_wheels.yaml --ref ${{ steps.bump.outputs.tag }}

{xpk-0.16.1 → xpk-0.17.1}/.github/workflows/reusable_build_wheel.yaml

@@ -29,6 +29,7 @@ jobs:
               with:
                   persist-credentials: false
                   fetch-depth: 0
+                  fetch-tags: true
             - name: Set up Python
               uses: actions/setup-python@v5
               with:

{xpk-0.16.1 → xpk-0.17.1}/.github/workflows/reusable_storage_create.yaml

@@ -18,10 +18,10 @@ on:
       cluster-name:
         type: string
         required: true
-      tpu-type:
-        description: 'TPU Type'
+      device-type:
+        description: 'Device Type'
         required: false
-        default: 'v4-8'
+        default: 'n2-standard-64-1'
         type: string
       zone:
         type: string
@@ -102,14 +102,14 @@ jobs:
       run: |
         kubectl get pv ${{inputs.storage-name}}-pv -oyaml | grep 'gcsfuseMetadataPrefetchOnMount: "true"' || (echo 'Metadata pre-population was not enabled' && exit 143)
     - name: Run workload to write file on filestore
-      run: xpk workload create --workload $STORAGE_WRITE_WORKLOAD --num-slices=1 --docker-image='marketplace.gcr.io/google/ubuntu2004' --command "mkdir -p /${{inputs.storage-type}}-test-mount-point/$RANDOM_SEED/ && echo 'Test text message' > /${{inputs.storage-type}}-test-mount-point/$RANDOM_SEED/test.txt || (echo 'Writing to filestore failed' && exit 143)" --cluster ${{inputs.cluster-name}} --tpu-type=${{inputs.tpu-type}} --zone ${{inputs.zone}}
+      run: xpk workload create --workload $STORAGE_WRITE_WORKLOAD --num-slices=1 --docker-image='marketplace.gcr.io/google/ubuntu2004' --command "mkdir -p /${{inputs.storage-type}}-test-mount-point/$RANDOM_SEED/ && echo 'Test text message' > /${{inputs.storage-type}}-test-mount-point/$RANDOM_SEED/test.txt || (echo 'Writing to filestore failed' && exit 143)" --cluster ${{inputs.cluster-name}} --device-type=${{inputs.device-type}} --zone ${{inputs.zone}}
     - name: Wait for writer workload completion and confirm it succeeded
       run: xpk workload list --cluster ${{inputs.cluster-name}} --zone=${{inputs.zone}} --wait-for-job-completion $STORAGE_WRITE_WORKLOAD --timeout 300
     - name: Delete the writer workload on the cluster
       if: always()
       run: xpk workload delete --workload $STORAGE_WRITE_WORKLOAD --cluster ${{inputs.cluster-name}} --zone=${{inputs.zone}}
     - name: Run workload to read file on filestore
-      run : xpk workload create --workload $STORAGE_READ_WORKLOAD --command "grep 'Test text message' /${{inputs.storage-type}}-test-mount-point/$RANDOM_SEED/test.txt || (echo 'Reading from filestore failed' && exit 143)" --cluster ${{inputs.cluster-name}} --tpu-type=${{inputs.tpu-type}} --zone ${{inputs.zone}}
+      run : xpk workload create --workload $STORAGE_READ_WORKLOAD --command "grep 'Test text message' /${{inputs.storage-type}}-test-mount-point/$RANDOM_SEED/test.txt || (echo 'Reading from filestore failed' && exit 143)" --cluster ${{inputs.cluster-name}} --device-type=${{inputs.device-type}} --zone ${{inputs.zone}}
     - name: Wait for reader workload completion and confirm it succeeded
       run: xpk workload list --cluster ${{inputs.cluster-name}} --zone=${{inputs.zone}} --wait-for-job-completion $STORAGE_READ_WORKLOAD --timeout 300
     - name: Delete the reader workload on the cluster
@@ -154,7 +154,7 @@ jobs:
     - name: Delete create-shell.exp file
       run: rm create-shell.exp
     - name: Run workload to delete file on filestore
-      run : xpk workload create --workload $STORAGE_DELETE_WORKLOAD --command "rm -rf /${{inputs.storage-type}}-test-mount-point/$RANDOM_SEED/test.txt || exit 143" --num-slices=1 --cluster ${{inputs.cluster-name}} --tpu-type=${{inputs.tpu-type}} --zone ${{inputs.zone}}
+      run : xpk workload create --workload $STORAGE_DELETE_WORKLOAD --command "rm -rf /${{inputs.storage-type}}-test-mount-point/$RANDOM_SEED/test.txt || exit 143" --num-slices=1 --cluster ${{inputs.cluster-name}} --device-type=${{inputs.device-type}} --zone ${{inputs.zone}}
     - name: Wait for delete workload completion and confirm it succeeded
       run: xpk workload list --cluster ${{inputs.cluster-name}} --zone=${{inputs.zone}} --wait-for-job-completion $STORAGE_DELETE_WORKLOAD --timeout 300
     - name: Delete the delete workload on the cluster

{xpk-0.16.1/src/xpk.egg-info → xpk-0.17.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xpk
-Version: 0.16.1
+Version: 0.17.1
 Summary: xpk helps Cloud developers to orchestrate training jobs on accelerators on GKE.
 Author-email: XPK team <xpk-code-reviewers@google.com>
 License: Apache-2.0
@@ -25,6 +25,7 @@ Requires-Dist: packaging==24.2
 Requires-Dist: google-cloud-filestore==1.12.0
 Requires-Dist: google-cloud-storage
 Requires-Dist: Jinja2==3.1.6
+Requires-Dist: urllib3<2.6.0
 Provides-Extra: dev
 Requires-Dist: pyink==24.3.0; extra == "dev"
 Requires-Dist: pylint>=2.6.0; extra == "dev"

{xpk-0.16.1 → xpk-0.17.1}/docs/installation.md

@@ -86,7 +86,7 @@ If you need to modify the source code or use the latest unreleased features:
 
 ```shell
 # 1. Clone the XPK repository
-git clone [https://github.com/google/xpk.git](https://github.com/google/xpk.git)
+git clone https://github.com/AI-Hypercomputer/xpk.git
 cd xpk
 
 # 2. Install dependencies and build
@@ -96,6 +96,8 @@ make install
 export PATH=$PATH:$PWD/bin
 ```
 
+*Note: Installing from source is recommended only for contributors and advanced users. Most users should install via PIP for the best stability.*
+
 **Persisting the PATH configuration:**
 To use `xpk` in future terminal sessions without re-running the export command, add the binary path to your shell configuration:
 

{xpk-0.16.1 → xpk-0.17.1}/goldens/Basic_cluster_create.txt

@@ -37,11 +37,11 @@ kubectl wait deployment/coredns --for=condition=Available=true --namespace=kube-
 [XPK] Task: `Determine current gke master version` is implemented by the following command not running since it is a dry run. 
 gcloud beta container clusters describe golden-cluster --location us-central1 --project golden-project --format="value(currentMasterVersion)"
 [XPK] Creating 1 node pool or pools of tpu7x-8
-We assume that the underlying system is: SystemCharacteristics(topology='2x2x1', vms_per_slice=1, gke_accelerator='tpu7x', gce_machine_type='tpu7x-standard-4t', chips_per_vm=4, accelerator_type=TPU, device_type='tpu7x-8', supports_sub_slicing=False, docker_platform=<DockerPlatform.AMD: 'linux/amd64'>, requires_workload_policy=False, gpu_config=None)
+We assume that the underlying system is: SystemCharacteristics(topology='2x2x1', vms_per_slice=1, gke_accelerator='tpu7x', gce_machine_type='tpu7x-standard-4t', chips_per_vm=4, accelerator_type=TPU, device_type='tpu7x-8', supports_sub_slicing=False, supports_super_slicing=False, docker_platform=<DockerPlatform.AMD: 'linux/amd64'>, requires_workload_policy=False, gpu_config=None)
 [XPK] Task: `Get All Node Pools` is implemented by the following command not running since it is a dry run. 
 gcloud beta container node-pools list --cluster golden-cluster --project=golden-project --location=us-central1 --format="csv[no-heading](name)"
 [XPK] Creating 1 node pool or pools of tpu7x-8
-Underlyingly, we assume that means: SystemCharacteristics(topology='2x2x1', vms_per_slice=1, gke_accelerator='tpu7x', gce_machine_type='tpu7x-standard-4t', chips_per_vm=4, accelerator_type=TPU, device_type='tpu7x-8', supports_sub_slicing=False, docker_platform=<DockerPlatform.AMD: 'linux/amd64'>, requires_workload_policy=False, gpu_config=None)
+Underlyingly, we assume that means: SystemCharacteristics(topology='2x2x1', vms_per_slice=1, gke_accelerator='tpu7x', gce_machine_type='tpu7x-standard-4t', chips_per_vm=4, accelerator_type=TPU, device_type='tpu7x-8', supports_sub_slicing=False, supports_super_slicing=False, docker_platform=<DockerPlatform.AMD: 'linux/amd64'>, requires_workload_policy=False, gpu_config=None)
 [XPK] Task: `Get Node Pool Zone` is implemented by the following command not running since it is a dry run. 
 gcloud beta container node-pools describe 0 --cluster golden-cluster --project=golden-project --location=us-central1 --format="value(locations)"
 [XPK] Task: `GKE Cluster Get ConfigMap` is implemented by the following command not running since it is a dry run. 
@@ -174,7 +174,7 @@ kubectl get deployment kueue-controller-manager -n kueue-system -o jsonpath='{.s
 kubectl apply --server-side --force-conflicts -f https://github.com/kubernetes-sigs/kueue/releases/download/v0.14.3/manifests.yaml
 [XPK] Task: `Wait for Kueue to be available` is implemented by the following command not running since it is a dry run. 
 kubectl wait deploy/kueue-controller-manager -n kueue-system --for=condition=available --timeout=10m
-[XPK] Temp file (ce52d2868b681f478f3f12e5696b1609e68b442a32f7f82603ba7064b825cf4f) content: 
+[XPK] Temp file (6083d72fc3ba2ac7d243c1269dd67717abd4086bf64e397e3a1737de415dd133) content: 
 
 apiVersion: kueue.x-k8s.io/v1beta1
 kind: ResourceFlavor
@@ -182,19 +182,6 @@ metadata:
   name: "1xtpu7x-8"
 spec:
   nodeLabels: {"cloud.google.com/gke-tpu-accelerator": "tpu7x", "cloud.google.com/gke-tpu-topology": "2x2x1"}
-  
----
-
-apiVersion: kueue.x-k8s.io/v1beta1
-kind: AdmissionCheck
-metadata:
-  name: dws-prov
-spec:
-  controllerName: kueue.x-k8s.io/provisioning-request
-  parameters:
-    apiGroup: kueue.x-k8s.io
-    kind: ProvisioningRequestConfig
-    name: dws-config
 ---
 apiVersion: kueue.x-k8s.io/v1beta1
 kind: ProvisioningRequestConfig
@@ -219,7 +206,6 @@ spec:
     withinClusterQueue: LowerPriority
   namespaceSelector: {} # match all.
   resourceGroups: [{'coveredResources': ['google.com/tpu'], 'flavors': [{'name': '1xtpu7x-8', 'resources': [{'name': 'google.com/tpu', 'nominalQuota': 4}]}]}]
-  
 ---
 apiVersion: kueue.x-k8s.io/v1beta1
 kind: LocalQueue
@@ -269,7 +255,7 @@ value: 1000
 globalDefault: false
 description: "Very High"
 [XPK] Task: `Applying Kueue Custom Resources` is implemented by the following command not running since it is a dry run. 
-kubectl apply -f ce52d2868b681f478f3f12e5696b1609e68b442a32f7f82603ba7064b825cf4f
+kubectl apply -f 6083d72fc3ba2ac7d243c1269dd67717abd4086bf64e397e3a1737de415dd133
 [XPK] Task: `Count total nodes` is implemented by the following command not running since it is a dry run. 
 kubectl get node --no-headers | wc -l
 [XPK] Try 1: Updating Kueue Controller Manager resources

{xpk-0.16.1 → xpk-0.17.1}/goldens/Cluster_create_for_multi-host_nodepool.txt

@@ -37,11 +37,11 @@ kubectl wait deployment/coredns --for=condition=Available=true --namespace=kube-
 [XPK] Task: `Determine current gke master version` is implemented by the following command not running since it is a dry run. 
 gcloud beta container clusters describe golden-cluster --location us-central1 --project golden-project --format="value(currentMasterVersion)"
 [XPK] Creating 1 node pool or pools of tpu7x-16
-We assume that the underlying system is: SystemCharacteristics(topology='2x2x2', vms_per_slice=2, gke_accelerator='tpu7x', gce_machine_type='tpu7x-standard-4t', chips_per_vm=4, accelerator_type=TPU, device_type='tpu7x-16', supports_sub_slicing=False, docker_platform=<DockerPlatform.AMD: 'linux/amd64'>, requires_workload_policy=True, gpu_config=None)
+We assume that the underlying system is: SystemCharacteristics(topology='2x2x2', vms_per_slice=2, gke_accelerator='tpu7x', gce_machine_type='tpu7x-standard-4t', chips_per_vm=4, accelerator_type=TPU, device_type='tpu7x-16', supports_sub_slicing=False, supports_super_slicing=False, docker_platform=<DockerPlatform.AMD: 'linux/amd64'>, requires_workload_policy=True, gpu_config=None)
 [XPK] Task: `Get All Node Pools` is implemented by the following command not running since it is a dry run. 
 gcloud beta container node-pools list --cluster golden-cluster --project=golden-project --location=us-central1 --format="csv[no-heading](name)"
 [XPK] Creating 1 node pool or pools of tpu7x-16
-Underlyingly, we assume that means: SystemCharacteristics(topology='2x2x2', vms_per_slice=2, gke_accelerator='tpu7x', gce_machine_type='tpu7x-standard-4t', chips_per_vm=4, accelerator_type=TPU, device_type='tpu7x-16', supports_sub_slicing=False, docker_platform=<DockerPlatform.AMD: 'linux/amd64'>, requires_workload_policy=True, gpu_config=None)
+Underlyingly, we assume that means: SystemCharacteristics(topology='2x2x2', vms_per_slice=2, gke_accelerator='tpu7x', gce_machine_type='tpu7x-standard-4t', chips_per_vm=4, accelerator_type=TPU, device_type='tpu7x-16', supports_sub_slicing=False, supports_super_slicing=False, docker_platform=<DockerPlatform.AMD: 'linux/amd64'>, requires_workload_policy=True, gpu_config=None)
 [XPK] Task: `Get Node Pool Zone` is implemented by the following command not running since it is a dry run. 
 gcloud beta container node-pools describe 0 --cluster golden-cluster --project=golden-project --location=us-central1 --format="value(locations)"
 [XPK] Task: `GKE Cluster Get ConfigMap` is implemented by the following command not running since it is a dry run. 
@@ -176,7 +176,7 @@ kubectl get deployment kueue-controller-manager -n kueue-system -o jsonpath='{.s
 kubectl apply --server-side --force-conflicts -f https://github.com/kubernetes-sigs/kueue/releases/download/v0.14.3/manifests.yaml
 [XPK] Task: `Wait for Kueue to be available` is implemented by the following command not running since it is a dry run. 
 kubectl wait deploy/kueue-controller-manager -n kueue-system --for=condition=available --timeout=10m
-[XPK] Temp file (f0a510ac08b9c6d8f549478c49836dca41a72a347c491acac1fa70272d531056) content: 
+[XPK] Temp file (b58f50dd88cb1211d51276b9b445f6bca02f0e97fa984656d47992aecd9322cc) content: 
 
 apiVersion: kueue.x-k8s.io/v1beta1
 kind: ResourceFlavor
@@ -184,19 +184,6 @@ metadata:
   name: "1xtpu7x-16"
 spec:
   nodeLabels: {"cloud.google.com/gke-tpu-accelerator": "tpu7x", "cloud.google.com/gke-tpu-topology": "2x2x2"}
-  
----
-
-apiVersion: kueue.x-k8s.io/v1beta1
-kind: AdmissionCheck
-metadata:
-  name: dws-prov
-spec:
-  controllerName: kueue.x-k8s.io/provisioning-request
-  parameters:
-    apiGroup: kueue.x-k8s.io
-    kind: ProvisioningRequestConfig
-    name: dws-config
 ---
 apiVersion: kueue.x-k8s.io/v1beta1
 kind: ProvisioningRequestConfig
@@ -221,7 +208,6 @@ spec:
     withinClusterQueue: LowerPriority
   namespaceSelector: {} # match all.
   resourceGroups: [{'coveredResources': ['google.com/tpu'], 'flavors': [{'name': '1xtpu7x-16', 'resources': [{'name': 'google.com/tpu', 'nominalQuota': 8}]}]}]
-  
 ---
 apiVersion: kueue.x-k8s.io/v1beta1
 kind: LocalQueue
@@ -271,7 +257,7 @@ value: 1000
 globalDefault: false
 description: "Very High"
 [XPK] Task: `Applying Kueue Custom Resources` is implemented by the following command not running since it is a dry run. 
-kubectl apply -f f0a510ac08b9c6d8f549478c49836dca41a72a347c491acac1fa70272d531056
+kubectl apply -f b58f50dd88cb1211d51276b9b445f6bca02f0e97fa984656d47992aecd9322cc
 [XPK] Task: `Count total nodes` is implemented by the following command not running since it is a dry run. 
 kubectl get node --no-headers | wc -l
 [XPK] Try 1: Updating Kueue Controller Manager resources

{xpk-0.16.1 → xpk-0.17.1}/goldens/Cluster_create_private.txt

@@ -41,13 +41,13 @@ kubectl wait deployment/coredns --for=condition=Available=true --namespace=kube-
 [XPK] Task: `Determine current gke master version` is implemented by the following command not running since it is a dry run. 
 gcloud beta container clusters describe golden-cluster-private --location us-central1 --project golden-project --format="value(currentMasterVersion)"
 [XPK] Creating 1 node pool or pools of v5p-8
-We assume that the underlying system is: SystemCharacteristics(topology='2x2x1', vms_per_slice=1, gke_accelerator='tpu-v5p-slice', gce_machine_type='ct5p-hightpu-4t', chips_per_vm=4, accelerator_type=TPU, device_type='v5p-8', supports_sub_slicing=False, docker_platform=<DockerPlatform.AMD: 'linux/amd64'>, requires_workload_policy=False, gpu_config=None)
+We assume that the underlying system is: SystemCharacteristics(topology='2x2x1', vms_per_slice=1, gke_accelerator='tpu-v5p-slice', gce_machine_type='ct5p-hightpu-4t', chips_per_vm=4, accelerator_type=TPU, device_type='v5p-8', supports_sub_slicing=False, supports_super_slicing=False, docker_platform=<DockerPlatform.AMD: 'linux/amd64'>, requires_workload_policy=False, gpu_config=None)
 [XPK] Task: `Get All Node Pools` is implemented by the following command not running since it is a dry run. 
 gcloud beta container node-pools list --cluster golden-cluster-private --project=golden-project --location=us-central1 --format="csv[no-heading](name)"
 [XPK] Task: `Describe reservation` is implemented by the following command not running since it is a dry run. 
 gcloud beta compute reservations describe golden-reservation --project=golden-project --zone=us-central1-a
 [XPK] Creating 1 node pool or pools of v5p-8
-Underlyingly, we assume that means: SystemCharacteristics(topology='2x2x1', vms_per_slice=1, gke_accelerator='tpu-v5p-slice', gce_machine_type='ct5p-hightpu-4t', chips_per_vm=4, accelerator_type=TPU, device_type='v5p-8', supports_sub_slicing=False, docker_platform=<DockerPlatform.AMD: 'linux/amd64'>, requires_workload_policy=False, gpu_config=None)
+Underlyingly, we assume that means: SystemCharacteristics(topology='2x2x1', vms_per_slice=1, gke_accelerator='tpu-v5p-slice', gce_machine_type='ct5p-hightpu-4t', chips_per_vm=4, accelerator_type=TPU, device_type='v5p-8', supports_sub_slicing=False, supports_super_slicing=False, docker_platform=<DockerPlatform.AMD: 'linux/amd64'>, requires_workload_policy=False, gpu_config=None)
 [XPK] Task: `Get Node Pool Zone` is implemented by the following command not running since it is a dry run. 
 gcloud beta container node-pools describe 0 --cluster golden-cluster-private --project=golden-project --location=us-central1 --format="value(locations)"
 [XPK] Task: `GKE Cluster Get ConfigMap` is implemented by the following command not running since it is a dry run. 
@@ -184,7 +184,7 @@ kubectl get deployment kueue-controller-manager -n kueue-system -o jsonpath='{.s
 kubectl apply --server-side --force-conflicts -f https://github.com/kubernetes-sigs/kueue/releases/download/v0.14.3/manifests.yaml
 [XPK] Task: `Wait for Kueue to be available` is implemented by the following command not running since it is a dry run. 
 kubectl wait deploy/kueue-controller-manager -n kueue-system --for=condition=available --timeout=10m
-[XPK] Temp file (1838a525f73d2cfd19aa616665e8b33fcc4ea10ba8d6015a9307109a6be6d372) content: 
+[XPK] Temp file (2e0015f210b664c3b767ae4e11af51387b01d4d6b36e20fecbdee137d3d2700b) content: 
 
 apiVersion: kueue.x-k8s.io/v1beta1
 kind: ResourceFlavor
@@ -192,28 +192,13 @@ metadata:
   name: "1xv5p-8"
 spec:
   nodeLabels: {"cloud.google.com/gke-tpu-accelerator": "tpu-v5p-slice", "cloud.google.com/gke-tpu-topology": "2x2x1"}
-  
 ---
-
 apiVersion: kueue.x-k8s.io/v1beta1
 kind: ResourceFlavor
 metadata:
   name: "cpu-user"
 spec:
   nodeLabels: {"cloud.google.com/gke-nodepool": "cpu-np"}
-  
----
-
-apiVersion: kueue.x-k8s.io/v1beta1
-kind: AdmissionCheck
-metadata:
-  name: dws-prov
-spec:
-  controllerName: kueue.x-k8s.io/provisioning-request
-  parameters:
-    apiGroup: kueue.x-k8s.io
-    kind: ProvisioningRequestConfig
-    name: dws-config
 ---
 apiVersion: kueue.x-k8s.io/v1beta1
 kind: ProvisioningRequestConfig
@@ -238,7 +223,6 @@ spec:
     withinClusterQueue: LowerPriority
   namespaceSelector: {} # match all.
   resourceGroups: [{'coveredResources': ['google.com/tpu'], 'flavors': [{'name': '1xv5p-8', 'resources': [{'name': 'google.com/tpu', 'nominalQuota': 4}]}]}, {'coveredResources': ['cpu', 'memory'], 'flavors': [{'name': 'cpu-user', 'resources': [{'name': 'cpu', 'nominalQuota': 480}, {'name': 'memory', 'nominalQuota': '2000G'}]}]}]
-  
 ---
 apiVersion: kueue.x-k8s.io/v1beta1
 kind: LocalQueue
@@ -288,7 +272,7 @@ value: 1000
 globalDefault: false
 description: "Very High"
 [XPK] Task: `Applying Kueue Custom Resources` is implemented by the following command not running since it is a dry run. 
-kubectl apply -f 1838a525f73d2cfd19aa616665e8b33fcc4ea10ba8d6015a9307109a6be6d372
+kubectl apply -f 2e0015f210b664c3b767ae4e11af51387b01d4d6b36e20fecbdee137d3d2700b
 [XPK] Task: `Count total nodes` is implemented by the following command not running since it is a dry run. 
 kubectl get node --no-headers | wc -l
 [XPK] Try 1: Updating Kueue Controller Manager resources