Skip to content

Commit 596a211

Browse files
authored
fix: Fix syntax errors (GoogleCloudPlatform#1316)
Fix syntax errors caused by GoogleCloudPlatform#1308.
1 parent 29eecef commit 596a211

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

dags/sparsity_diffusion_devx/maxtext_moe_tpu_e2e.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -161,7 +161,7 @@ def convert_checkpoint_and_run_training(
161161
test_name=test_name,
162162
run_model_cmds=(
163163
f"export BASE_OUTPUT_PATH=$GCS_OUTPUT; bash tests/end_to_end/"
164-
f"{test_scripts_details_list[0]['script_name']}.sh",
164+
f"{test_scripts_details_list[0]['script_name']}.sh"
165165
),
166166
docker_image=docker_image_config,
167167
test_owner=test_owner.SHUNING_J,
@@ -172,11 +172,11 @@ def convert_checkpoint_and_run_training(
172172
test_name=test_name,
173173
run_model_cmds=(
174174
f"export BASE_OUTPUT_PATH=$GCS_OUTPUT; bash tests/end_to_end/"
175-
f"{test_scripts_details_list[1]['script_name']}.sh",
175+
f"{test_scripts_details_list[1]['script_name']}.sh"
176176
),
177177
docker_image=docker_image_config,
178178
test_owner=test_owner.SHUNING_J,
179-
cluster=test_scripts_details[1]["cluster"],
179+
cluster=test_scripts_details_list[1]["cluster"],
180180
).run(gcs_location=shared_gcs_location)
181181
return conversion_cpu, training_tpu_task
182182

0 commit comments

Comments
 (0)