Skip to content

Commit c3b733f

Browse files
authored
fix: Adjust schedule for pw_mcjax_benchmark_recipe (GoogleCloudPlatform#1108)
This change adjusts the schedule for the `pw_mcjax_benchmark_recipe` DAG to: - Restrict execution to the PROD Composer environment (prevent non-PROD triggers). - Run outside peak hours to keep cluster capacity available.
1 parent 3acf653 commit c3b733f

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

dags/maxtext_pathways/pw_mcjax_dags.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
from airflow.providers.google.cloud.operators.kubernetes_engine import GKEStartPodOperator
2828
from kubernetes.client import models as k8s
2929

30+
from dags import composer_env
3031
from dags.common import test_owner
3132
from dags.maxtext_pathways.configs import commands as cmds
3233
from dags.maxtext_pathways.configs import parameters as ui_params
@@ -261,7 +262,7 @@ def wait_workload_complete(
261262
with models.DAG(
262263
dag_id=RECIPE_NAME,
263264
start_date=datetime.datetime(2025, 1, 1),
264-
schedule_interval="0 10 * * *",
265+
schedule_interval="0 21 * * *" if composer_env.is_prod_env() else None,
265266
catchup=False,
266267
default_args={
267268
"retries": 0,

0 commit comments

Comments
 (0)