From a74c590dc9bd0fb95663648e74f73f996a611ed7 Mon Sep 17 00:00:00 2001 From: Luke Baumann Date: Wed, 29 Jul 2026 12:24:00 -0700 Subject: [PATCH] pathwaysutils: Add elastic_slices validation to JobSet generator. PiperOrigin-RevId: 956025079 --- pathwaysutils/experimental/gke/jobset.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/pathwaysutils/experimental/gke/jobset.py b/pathwaysutils/experimental/gke/jobset.py index 9cd5470..28bee85 100644 --- a/pathwaysutils/experimental/gke/jobset.py +++ b/pathwaysutils/experimental/gke/jobset.py @@ -159,6 +159,11 @@ def __init__( self._labels = dict(labels) if labels else {} self._annotations = dict(annotations) if annotations else {} + if elastic_slices > num_slices: + raise ValueError( + f"elastic_slices ({elastic_slices}) cannot be greater than " + f"num_slices ({num_slices})" + ) tpu_version = MACHINE_TYPE_TO_TPU_VERSION_MAP.get(tpu_type.lower()) if not tpu_version: raise ValueError(f"Unsupported TPU type: {tpu_type}")