Add support for tpu v6e quota check (#4111)

rayandasoriya · Rayan Dasoriya · web-flow · commit e563a6611400 · 2025-06-17T13:01:19.000Z
Co-authored-by: Rayan Dasoriya &lt;dasoriya@google.com&gt;
diff --git a/community-content/vertex_model_garden/model_oss/notebook_util/common_util.py b/community-content/vertex_model_garden/model_oss/notebook_util/common_util.py
@@ -564,6 +564,7 @@ def get_resource_id(
       "NVIDIA_H100_MEGA_80GB": "nvidia_h100_mega_gpus",
       "NVIDIA_H200_141GB": "nvidia_h200_gpus",
       "NVIDIA_TESLA_T4": "nvidia_t4_gpus",
+      "TPU_V6e": "tpu_v6e",
       "TPU_V5e": "tpu_v5e",
       "TPU_V3": "tpu_v3",
   }

Original file line number	Diff line number	Diff line change
`@@ -564,6 +564,7 @@ def get_resource_id(`
`564`	`564`	`"NVIDIA_H100_MEGA_80GB": "nvidia_h100_mega_gpus",`
`565`	`565`	`"NVIDIA_H200_141GB": "nvidia_h200_gpus",`
`566`	`566`	`"NVIDIA_TESLA_T4": "nvidia_t4_gpus",`
	`567`	`+ "TPU_V6e": "tpu_v6e",`
`567`	`568`	`"TPU_V5e": "tpu_v5e",`
`568`	`569`	`"TPU_V3": "tpu_v3",`
`569`	`570`	`}`