diff --git a/packages/k8s/src/hooks/prepare-job.ts b/packages/k8s/src/hooks/prepare-job.ts index ea430ccf..84aa450d 100644 --- a/packages/k8s/src/hooks/prepare-job.ts +++ b/packages/k8s/src/hooks/prepare-job.ts @@ -164,10 +164,13 @@ 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(