diff --git a/pathwaysutils/experimental/gke/jobset.py b/pathwaysutils/experimental/gke/jobset.py index 28bee85..ecfbaf7 100644 --- a/pathwaysutils/experimental/gke/jobset.py +++ b/pathwaysutils/experimental/gke/jobset.py @@ -407,6 +407,7 @@ def _build_worker_job_template( client.V1EnvVar(name="TF_CPP_MIN_LOG_LEVEL", value="0"), client.V1EnvVar(name="XCLOUD_ENVIRONMENT", value="GCP"), client.V1EnvVar(name="MEGASCALE_GRPC_ENABLE_XOR_TRACER", value="false"), + client.V1EnvVar(name="MEGASCALE_PORT", value="8080"), client.V1EnvVar( name="MEGASCALE_NUM_SLICES", value_from=client.V1EnvVarSource(