Skip to content

Commit 22f515f

Browse files
Fix: Update cluster locations for map_reproducibility tests (GoogleCloudPlatform#751)
This commit updates the cluster names and regions for specific hardware configurations used in map_reproducibility tests. The following changes were made in `dags/map_reproducibility/utils/common_utils.py`: - For "a3ultra" hardware, the cluster is updated to "imo-a3ultra" in "europe-west1". - For "a4" hardware, the cluster is updated to "gke-a4-sbrg1" in "us-east4". Co-authored-by: google-labs-jules[bot] <161369871+google-labs-jules[bot]@users.noreply.github.com>
1 parent c1b93b9 commit 22f515f

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

dags/map_reproducibility/utils/common_utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1056,9 +1056,9 @@ def get_cluster(hardware: str = "a3ultra"):
10561056
if hardware == "a3mega":
10571057
return "a3plus-benchmark", "australia-southeast1"
10581058
if hardware == "a3ultra":
1059-
return "gke-a3ultra-bm-map-3", "europe-west1"
1059+
return "imo-a3ultra", "europe-west1"
10601060
if hardware == "a4":
1061-
return "gke-a4-shared", "us-central1"
1061+
return "gke-a4-sbrg1", "us-east4"
10621062

10631063

10641064
def get_scheduled_time(hardware: str, model: str, framework: str):

0 commit comments

Comments
 (0)