We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent b1ee705 commit 61058a2Copy full SHA for 61058a2
1 file changed
dags/map_reproducibility/utils/common_utils.py
@@ -411,12 +411,8 @@ def helm_apply_cmds_internal_run(
411
bucket_name=BUCKET_NAME,
412
):
413
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}"
+ if framework == "maxtext":
+ gcs_cmd += f" --set volumes.gcsMounts[0].bucketName={bucket_name} "
420
421
cluster_cmd = ""
422
if framework == "nemo" and hypercomputer == "a3ultra":
0 commit comments