Update TPU type validation: rename tpuv5p to tpuv5

Pathways-on-Cloud Team · copybara-github · commit 226c433d5877 · 2025-12-15T10:54:16.000-08:00
The regular expression for validating TPU instance strings is updated to allow 'tpuv5' in addition to 'tpuv5e' and 'tpuv6e', replacing 'tpuv5p'. Pathways uses 'tpuv5' for TPU v5p.

PiperOrigin-RevId: 843829112
diff --git a/pathwaysutils/experimental/shared_pathways_service/validators.py b/pathwaysutils/experimental/shared_pathways_service/validators.py
@@ -51,7 +51,7 @@ def _validate_tpu_supported(tpu_instance_with_topology: str) -> None:
   # tpuv6e:2x4 -> type='tpuv6e', topology='2x4'
   # tpuv5p:2x2x1 -> type='tpuv5p', topology='2x2x1'
   match = re.match(
-      r"^(?:tpuv(?:5e|5p|6e)):(?P<topology>\d+(?:x\d+){1,2})$",
+      r"^(?:tpuv(?:5e|5|6e)):(?P<topology>\d+(?:x\d+){1,2})$",
       tpu_instance_with_topology,
   )
 

Original file line number	Diff line number	Diff line change
`@@ -51,7 +51,7 @@ def _validate_tpu_supported(tpu_instance_with_topology: str) -> None:`
`51`	`51`	`# tpuv6e:2x4 -> type='tpuv6e', topology='2x4'`
`52`	`52`	`# tpuv5p:2x2x1 -> type='tpuv5p', topology='2x2x1'`
`53`	`53`	`match = re.match(`
`54`		`- r"^(?:tpuv(?:5e\|5p\|6e)):(?P<topology>\d+(?:x\d+){1,2})$",`
	`54`	`+ r"^(?:tpuv(?:5e\|5\|6e)):(?P<topology>\d+(?:x\d+){1,2})$",`
`55`	`55`	`tpu_instance_with_topology,`
`56`	`56`	`)`
`57`	`57`