Skip to content

Commit e0bf668

Browse files
update maxtext_end_to_end dag (GoogleCloudPlatform#1268)
TAG=agy CONV=4f66f305-cac9-4a35-946d-db30914c228d
1 parent 6a3ebf1 commit e0bf668

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

dags/multipod/maxtext_end_to_end.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -72,8 +72,9 @@
7272
"gemma3-4b": {
7373
"owner": test_owner.HENGTAO_G,
7474
"commands": [
75-
"bash tests/end_to_end/tpu/gemma3/4b/test_gemma3_to_mt.sh",
76-
"bash tests/end_to_end/tpu/gemma3/4b/test_gemma3_to_hf.sh",
75+
"export RUN_NAME=$(date +%Y-%m-%d-%H-%M-%S)",
76+
"bash tests/end_to_end/tpu/gemma3/4b/test_gemma3_to_mt.sh $RUN_NAME",
77+
"bash tests/end_to_end/tpu/gemma3/4b/test_gemma3.sh $RUN_NAME",
7778
],
7879
},
7980
"qwen3-4b": {

0 commit comments

Comments
 (0)