fix: Update timeout for jobset_ttr_drain_restart to 90 minutes

Author: lechipai0209

dags/common/scheduling_helper/scheduling_helper.py

@@ -71,7 +71,7 @@ class DayOfWeek(enum.Enum):
         "tpu_sdk_monitoring_validation": DefaultTimeout,
         "jobset_ttr_kill_process": dt.timedelta(minutes=90),
         "jobset_uptime_validation": dt.timedelta(minutes=90),
-        "jobset_ttr_drain_restart": DefaultTimeout,
+        "jobset_ttr_drain_restart": dt.timedelta(minutes=90),
         "tpu_info_metrics_verification": DefaultTimeout,
         "jobset_ttr_node_reboot": dt.timedelta(minutes=90),
     },