Fixit: Removing redundant tests (#559)

GoogleCloudPlatform · Jan 15, 2025 · cbb3a97 · cbb3a97
1 parent bb82d01
commit cbb3a97
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 10 deletions.
diff --git a/dags/sparsity_diffusion_devx/jax_stable_stack_gpu_e2e.py b/dags/sparsity_diffusion_devx/jax_stable_stack_gpu_e2e.py
@@ -68,15 +68,6 @@
 
   for model, (test_script, nnodes) in test_models_gpu.items():
     for mode, image in docker_images:
-      stable_a3_gpu = config.get_gpu_gke_test_config(
-          time_out_in_min=300,
-          test_name=f"maxtext-stable-stack-{mode.value}-{model}",
-          run_model_cmds=(test_script,),
-          num_slices=nnodes,
-          cluster=XpkClusters.GPU_A3_CLUSTER,
-          docker_image=image.value,
-          test_owner=test_owner.PARAM_B,
-      ).run_with_quarantine(quarantine_task_group)
       stable_a3plus_gpu = config.get_gpu_gke_test_config(
           time_out_in_min=300,
           test_name=f"maxtext-stable-stack-{mode.value}-{model}",

diff --git a/dags/sparsity_diffusion_devx/project_bite_gpu_e2e.py b/dags/sparsity_diffusion_devx/project_bite_gpu_e2e.py
@@ -44,7 +44,6 @@
 
   axlearn_test_configs = {
       # accelerator: list of slices to test
-      "a3": [1],
       "a3plus": [1, 2],
   }