diff --git a/CHANGELOG.md b/CHANGELOG.md index 2137e8b..cbe295a 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -9,6 +9,10 @@ use patch releases for compatibility fixes instead. ## Unreleased +### Added + +- Added `TaskSpec.propagate_preemption` field. + ## [v1.27.0](https://github.com/allenai/beaker-py/releases/tag/v1.27.0) - 2024-05-31 ### Added diff --git a/beaker/data_model/experiment_spec.py b/beaker/data_model/experiment_spec.py index f20f9f3..ef132f3 100644 --- a/beaker/data_model/experiment_spec.py +++ b/beaker/data_model/experiment_spec.py @@ -409,6 +409,11 @@ class TaskSpec(BaseModel, frozen=False): Determines if whole experiment should fail if this task failures. """ + propagate_preemption: Optional[bool] = None + """ + Determines if all tasks should be preempted if this one task is. + """ + synchronized_start_timeout: Optional[int] = None """ If set, jobs in the replicated task will wait to start, up to the specified timeout,