Update candidate jaii dag to schedule=None (GoogleCloudPlatform#921)

* Update candidate jaii DAG to manual trigger only

* fixed error in docker image value passed

Author: Rohan-Bierneni

dags/sparsity_diffusion_devx/jax_ai_image_candidate_tpu_e2e.py

@@ -38,6 +38,7 @@
     ],
     start_date=datetime.datetime(2025, 7, 24),
     catchup=False,
+    schedule=None,
 ) as dag:
   current_datetime = config.get_current_datetime()
   maxtext_test_configs = {
@@ -58,12 +59,12 @@
 
   maxtext_docker_images = [(
       SetupMode.STABLE,
-      DockerImage.MAXTEXT_TPU_JAX_STABLE_STACK_CANDIDATE,
+      "gcr.io/tpu-prod-env-multipod/maxtext_stable_stack_candidate:latest",
   )]
 
   maxdiffusion_docker_images = [(
       SetupMode.STABLE,
-      DockerImage.MAXDIFFUSION_TPU_JAX_STABLE_STACK_CANDIDATE,
+      "gcr.io/tpu-prod-env-multipod/maxdiffusion_stable_stack_candidate:latest",
   )]
 
   for accelerator, slices in maxtext_test_configs.items():
@@ -84,7 +85,7 @@
                 f"base_output_directory={gcs_bucket.BASE_OUTPUT_DIR}/maxtext/jax-stable-stack/automated/{current_datetime}",
             ),
             test_name=f"maxtext-jax-stable-stack-{mode.value}-{accelerator}-{slice_num}x",
-            docker_image=image.value,
+            docker_image=image,
             test_owner=test_owner.ROHAN_B,
         ).run_with_quarantine(quarantine_task_group)
 
@@ -108,6 +109,6 @@
                 f"output_dir={gcs_bucket.BASE_OUTPUT_DIR}/maxdiffusion-jax-stable-stack-{mode.value}-{accelerator}-{slice_num}/automated/{current_datetime}",
             ),
             test_name=f"maxdiffusion-jax-stable-stack-sdxl-{mode.value}-{accelerator}-{slice_num}x",
-            docker_image=image.value,
+            docker_image=image,
             test_owner=test_owner.ROHAN_B,
         ).run_with_quarantine(quarantine_task_group)