diff --git a/dags/mantaray/run_mantaray_jobs.py b/dags/mantaray/run_mantaray_jobs.py index 3e9b861c8..20c71548c 100644 --- a/dags/mantaray/run_mantaray_jobs.py +++ b/dags/mantaray/run_mantaray_jobs.py @@ -41,7 +41,7 @@ # merge all PyTorch/XLA tests ino one Dag with models.DAG( dag_id="pytorch_xla_model_regression_test_on_trillium", - schedule="0 0 * * *", # everyday at midnight # job["schedule"], + schedule="0 0 * * *" if composer_env.is_prod_env() else None, tags=["mantaray", "pytorchxla", "xlml"], start_date=datetime.datetime(2024, 4, 22), catchup=False,