Merge pull request #271 from allenai/michaelw/add-sync-start

Add `synchronized_start_timeout` flag
allenai · Apr 30, 2024 · deeb147 · deeb147
2 parents aa00c73 + 070d156
commit deeb147
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -9,6 +9,10 @@ use patch releases for compatibility fixes instead.
 
 ## Unreleased
 
+### Added
+
+- Added `synchronized_start_timeout` field to `TaskSpec`.
+
 ## [v1.26.6](https://github.com/allenai/beaker-py/releases/tag/v1.26.6) - 2024-04-24
 
 ### Added

diff --git a/beaker/data_model/experiment_spec.py b/beaker/data_model/experiment_spec.py
@@ -403,6 +403,13 @@ class TaskSpec(BaseModel, frozen=False):
     Determines if whole experiment should fail if this task failures.
     """
 
+    synchronized_start_timeout: Optional[str] = None
+    """
+    If set, jobs in the replicated task will wait to start, up to the specified timeout, 
+    until all other jobs are also ready. If the timeout is reached, the job will be canceled.
+    Must be greater than zero and less than or equal to 48 hours.
+    """
+
     @classmethod
     def new(
         cls,