feat: add dispatch_deadline parameter to tasks_v2.Task

flamingo-run · Jan 23, 2025 · a5ce055 · a5ce055
1 parent 933a112
commit a5ce055
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/gcp_pilot/tasks.py b/gcp_pilot/tasks.py
@@ -46,6 +46,7 @@ def push(
         use_oidc_auth: bool = True,
         content_type: str | None = None,
         headers: dict[str, str] | None = None,
+        dispatch_deadline: int | None = None,
     ) -> tasks_v2.Task:
         queue_path = self.client.queue_path(
             project=project_id or self.project_id,
@@ -81,6 +82,9 @@ def push(
             ),
         )
 
+        if dispatch_deadline is not None:
+            task.dispatch_deadline = dispatch_deadline
+
         if delay_in_seconds:
             target_date = datetime.now(tz=UTC) + timedelta(seconds=delay_in_seconds)
             timestamp = timestamp_pb2.Timestamp()