From e776b5de8b0559499d1fa81c1e12c6a53a5ceac2 Mon Sep 17 00:00:00 2001 From: Luke Baumann Date: Mon, 27 Jul 2026 16:49:44 -0700 Subject: [PATCH] pathwaysutils: Remove hostNetwork configuration from PathwaysJobSet pod specifications. PiperOrigin-RevId: 954902732 --- pathwaysutils/experimental/gke/jobset.py | 4 ---- pathwaysutils/test/experimental/gke/jobset_test.py | 8 ++++---- .../experimental/gke/testdata/model_lite_tpuv5e_4x8.yaml | 4 ---- 3 files changed, 4 insertions(+), 12 deletions(-) diff --git a/pathwaysutils/experimental/gke/jobset.py b/pathwaysutils/experimental/gke/jobset.py index c29781e..cedee48 100644 --- a/pathwaysutils/experimental/gke/jobset.py +++ b/pathwaysutils/experimental/gke/jobset.py @@ -320,8 +320,6 @@ def _build_head_job_template( containers.append(proxy_container) head_pod_spec = client.V1PodSpec( - host_network=True, - dns_policy="ClusterFirstWithHostNet", containers=containers, restart_policy="Never", ) @@ -466,8 +464,6 @@ def _build_worker_job_template( ), ) ], - host_network=True, - dns_policy="ClusterFirstWithHostNet", restart_policy="OnFailure", ) if termination_grace_period_seconds is not None: diff --git a/pathwaysutils/test/experimental/gke/jobset_test.py b/pathwaysutils/test/experimental/gke/jobset_test.py index b414db2..cc2fea6 100644 --- a/pathwaysutils/test/experimental/gke/jobset_test.py +++ b/pathwaysutils/test/experimental/gke/jobset_test.py @@ -149,8 +149,8 @@ def test_headless_head_job_pod_spec(self): self.assertIn("pathways-head", helper.jobs) self.assertEqual(helper.jobs["pathways-head"]["replicas"], 1) pod_spec = helper.pod_specs["pathways-head"] - self.assertTrue(pod_spec["hostNetwork"]) - self.assertEqual(pod_spec["dnsPolicy"], "ClusterFirstWithHostNet") + self.assertNotIn("hostNetwork", pod_spec) + self.assertNotIn("dnsPolicy", pod_spec) self.assertEqual(pod_spec["restartPolicy"], "Never") def test_headless_head_job_containers(self): @@ -207,8 +207,8 @@ def test_worker_job_pod_spec(self): helper = JobSetManifestHelper(config) pod_spec = helper.pod_specs["pathways-worker"] - self.assertTrue(pod_spec["hostNetwork"]) - self.assertEqual(pod_spec["dnsPolicy"], "ClusterFirstWithHostNet") + self.assertNotIn("hostNetwork", pod_spec) + self.assertNotIn("dnsPolicy", pod_spec) self.assertEqual(pod_spec["restartPolicy"], "OnFailure") self.assertEqual(pod_spec["terminationGracePeriodSeconds"], 60) diff --git a/pathwaysutils/test/experimental/gke/testdata/model_lite_tpuv5e_4x8.yaml b/pathwaysutils/test/experimental/gke/testdata/model_lite_tpuv5e_4x8.yaml index ad72f17..e84e4c8 100644 --- a/pathwaysutils/test/experimental/gke/testdata/model_lite_tpuv5e_4x8.yaml +++ b/pathwaysutils/test/experimental/gke/testdata/model_lite_tpuv5e_4x8.yaml @@ -73,8 +73,6 @@ spec: volumeMounts: - mountPath: /tmp name: shared-tmp - dnsPolicy: ClusterFirstWithHostNet - hostNetwork: true initContainers: - args: - --server_port=29001 @@ -209,8 +207,6 @@ spec: volumeMounts: - mountPath: /tmp name: shared-tmp - dnsPolicy: ClusterFirstWithHostNet - hostNetwork: true nodeSelector: cloud.google.com/gke-tpu-accelerator: tpu-v5-lite-podslice cloud.google.com/gke-tpu-topology: 4x8