Skip to content

Commit 61058a2

Browse files
authored
fix bucket name not found issue (#730)
1 parent b1ee705 commit 61058a2

1 file changed

Lines changed: 2 additions & 6 deletions

File tree

dags/map_reproducibility/utils/common_utils.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -411,12 +411,8 @@ def helm_apply_cmds_internal_run(
411411
bucket_name=BUCKET_NAME,
412412
):
413413
gcs_cmd = ""
414-
if hypercomputer in ("a3ultra", "a4"):
415-
if framework != "maxtext" and kueue_name:
416-
gcs_cmd = f" --set queue={kueue_name}"
417-
gcs_cmd += f" --set volumes.gcsMounts[0].bucketName={bucket_name}"
418-
else:
419-
gcs_cmd = f" --set workload.gcsBucketForDataCataPath={bucket_name}"
414+
if framework == "maxtext":
415+
gcs_cmd += f" --set volumes.gcsMounts[0].bucketName={bucket_name} "
420416

421417
cluster_cmd = ""
422418
if framework == "nemo" and hypercomputer == "a3ultra":

0 commit comments

Comments
 (0)