Only run gke mxla benchmark (#536)

* Only run gke mxla benchmark
GoogleCloudPlatform · Jan 10, 2025 · fbb96bf · fbb96bf
1 parent d628936
commit fbb96bf
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 147 deletions.
diff --git a/dags/multipod/mxla_maxtext_nightly.py b/dags/multipod/mxla_maxtext_nightly.py
diff --git a/dags/multipod/mxla_maxtext_nightly_gke.py b/dags/multipod/mxla_maxtext_nightly_gke.py
@@ -24,8 +24,7 @@
 from dags.multipod.configs import gke_config
 
 # Run once a day at 9 am UTC (1 am PST)
-# Pause test on GKE
-SCHEDULED_TIME = None
+SCHEDULED_TIME = "0 9 * * *" if composer_env.is_prod_env() else None
 
 with models.DAG(
     dag_id="mxla_maxtext_nightly_gke",