diff --git a/pathwaysutils/experimental/shared_pathways_service/validators.py b/pathwaysutils/experimental/shared_pathways_service/validators.py index 0276d5a..f5a6216 100644 --- a/pathwaysutils/experimental/shared_pathways_service/validators.py +++ b/pathwaysutils/experimental/shared_pathways_service/validators.py @@ -49,7 +49,7 @@ def _validate_tpu_supported(tpu_instance_with_topology: str) -> None: # Regex to extract TPU type and topology. # Examples: # tpuv6e:2x4 -> type='tpuv6e', topology='2x4' - # tpuv5p:2x2x1 -> type='tpuv5p', topology='2x2x1' + # tpuv5:2x2x1 -> type='tpuv5', topology='2x2x1' match = re.match( r"^(?:tpuv(?:5e|5|6e)):(?P\d+(?:x\d+){1,2})$", tpu_instance_with_topology,