File tree Expand file tree Collapse file tree
dags/sparsity_diffusion_devx Expand file tree Collapse file tree Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments