diff --git a/axlearn/cloud/gcp/job.py b/axlearn/cloud/gcp/job.py index 5e08ba64..6fca2b50 100644 --- a/axlearn/cloud/gcp/job.py +++ b/axlearn/cloud/gcp/job.py @@ -538,20 +538,6 @@ def _build_pod(self) -> Nested[Any]: PRE_PROVISIONER_LABEL: cfg.name, } ) - else: - # Used by GCP auto-provisioner. - selector.update( - { - # NOTE: This is an arbitrary key, with a value that must be unique to the - # jobset. This forces the jobset to be associated with its own node pool; - # without this, the TPU provisioner may create a node pool and the scheduler may - # schedule a different jobset onto the node pool, which can cause conflicts if - # the original jobset attempts to restart (node pool conflict). This is more - # reliable at the moment but doesn't take advantage of node pool sharing. GCP is - # working on a fix. - "provisioner-nodepool-id": cfg.name, - } - ) annotations.update( {