add TaskSpec.propagate_preemption field

allenai · May 31, 2024 · 1c1054b · 1c1054b
1 parent ef85f3f
commit 1c1054b
Show file tree

Hide file tree

Showing 2 changed files with 9 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 `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
@@ -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,