Fix the incomplete checkpoint error (GoogleCloudPlatform#1073)

This change fixes the `ValueError: Found incomplete checkpoint error` by setting the dependency in the tasks loop.

Author: RUEI4341

dags/multipod/maxtext_sft_trainer.py

@@ -47,7 +47,7 @@
       (SetupMode.STABLE, DockerImage.MAXTEXT_TPU_JAX_STABLE_STACK_CANDIDATE),
       (SetupMode.NIGHTLY, DockerImage.MAXTEXT_TPU_STABLE_STACK_NIGHTLY_JAX),
   ]
-
+  test = []
   for mode, image in docker_images:
     command = (
         f'export HF_TOKEN={HF_TOKEN}',
@@ -67,3 +67,6 @@
         docker_image=image.value,
         test_owner=test_owner.SURBHI_J,
     ).run()
+    test.append(maxtext_v4_configs_test)
+  for i in range(len(test) - 1):
+    test[i] >> test[i + 1]