From 85ea0f2e6e1db21dc13871b0c1c2506ef741ecda Mon Sep 17 00:00:00 2001 From: Quoc Truong Date: Thu, 16 Oct 2025 15:47:43 -0700 Subject: [PATCH 1/2] Fix bug with TPU request --- packages/k8s/src/hooks/prepare-job.ts | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/packages/k8s/src/hooks/prepare-job.ts b/packages/k8s/src/hooks/prepare-job.ts index ea430ccf..0e5bdb13 100644 --- a/packages/k8s/src/hooks/prepare-job.ts +++ b/packages/k8s/src/hooks/prepare-job.ts @@ -164,10 +164,11 @@ export function processServiceContainers( ) }) - const tpuRequestingContainers = services.filter( + const tpuRequestingContainers = serviceContainers.filter( service => service.resources?.limits && service.resources.limits['google.com/tpu'] ) + core.debug(`There are ${tpuRequestingContainers.length} service container requesting for TPU's.`) if (tpuRequestingContainers.length > 1) { throw new Error( From 2324c212cf8824888ed461a5957b772490e754c5 Mon Sep 17 00:00:00 2001 From: Quoc Truong Date: Thu, 16 Oct 2025 15:53:08 -0700 Subject: [PATCH 2/2] Fix formatting --- packages/k8s/src/hooks/prepare-job.ts | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/packages/k8s/src/hooks/prepare-job.ts b/packages/k8s/src/hooks/prepare-job.ts index 0e5bdb13..84aa450d 100644 --- a/packages/k8s/src/hooks/prepare-job.ts +++ b/packages/k8s/src/hooks/prepare-job.ts @@ -168,7 +168,9 @@ export function processServiceContainers( service => service.resources?.limits && service.resources.limits['google.com/tpu'] ) - core.debug(`There are ${tpuRequestingContainers.length} service container requesting for TPU's.`) + core.debug( + `There are ${tpuRequestingContainers.length} service container requesting for TPU's.` + ) if (tpuRequestingContainers.length > 1) { throw new Error(