Fix bug with TPU request

quoctruong · quoctruong · commit 85ea0f2e6e1d · 2025-10-16T15:47:43.000-07:00
diff --git 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(

Original file line number	Diff line number	Diff line change
`@@ -164,10 +164,11 @@ export function processServiceContainers(`
`164`	`164`	`)`
`165`	`165`	`})`
`166`	`166`
`167`		`- const tpuRequestingContainers = services.filter(`
	`167`	`+ const tpuRequestingContainers = serviceContainers.filter(`
`168`	`168`	`service =>`
`169`	`169`	`service.resources?.limits && service.resources.limits['google.com/tpu']`
`170`	`170`	`)`
	`171`	+ core.debug(`There are ${tpuRequestingContainers.length} service container requesting for TPU's.`)
`171`	`172`
`172`	`173`	`if (tpuRequestingContainers.length > 1) {`
`173`	`174`	`throw new Error(`