From 070d156ca70e2acd59980ece4756b84420f00467 Mon Sep 17 00:00:00 2001 From: michael wilson Date: Mon, 29 Apr 2024 15:19:13 -0700 Subject: [PATCH] Add synchronized_start_timeout flag --- CHANGELOG.md | 4 ++++ beaker/data_model/experiment_spec.py | 7 +++++++ 2 files changed, 11 insertions(+) diff --git a/CHANGELOG.md b/CHANGELOG.md index 0f5a432..f30e3f7 100644 --- 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 index 1197722..ce52894 100644 --- 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,