Fix test dependency issue

GoogleCloudPlatform · Nov 15, 2024 · 79e025f · 79e025f
1 parent 5af6bd7
commit 79e025f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/dags/multipod/maxtext_end_to_end.py b/dags/multipod/maxtext_end_to_end.py
@@ -162,4 +162,4 @@ def convert_checkpoint_and_run_training(
 
     # stable_cpu >> stable_tpu >> nightly_cpu >> nightly_tpu
     for i in range(len(tests) - 1):
-      tests[i] << tests[i + 1]
+      tests[i] >> tests[i + 1]
diff --git a/dags/sparsity_diffusion_devx/maxtext_moe_tpu_e2e.py b/dags/sparsity_diffusion_devx/maxtext_moe_tpu_e2e.py
@@ -145,4 +145,4 @@ def convert_checkpoint_and_run_training(
 
     # stable_cpu >> stable_tpu >> nightly_cpu >> nightly_tpu
     for i in range(len(tests) - 1):
-      tests[i] << tests[i + 1]
+      tests[i] >> tests[i + 1]