diff --git a/CHANGELOG.md b/CHANGELOG.md index ba00746..f4d0702 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -9,6 +9,10 @@ use patch releases for compatibility fixes instead. ## Unreleased +### Added + +- Added new field `TaskContext.preemptible`. + ## [v1.26.10](https://github.com/allenai/beaker-py/releases/tag/v1.26.10) - 2024-05-02 ### Added diff --git a/beaker/data_model/experiment_spec.py b/beaker/data_model/experiment_spec.py index 7674956..3d94c44 100644 --- a/beaker/data_model/experiment_spec.py +++ b/beaker/data_model/experiment_spec.py @@ -274,6 +274,11 @@ class TaskContext(BaseModel, frozen=False): Tasks with higher priority are placed ahead of tasks with lower priority in the queue. """ + preemptible: Optional[bool] = None + """ + Whether or not a job is marked as preemptible. + """ + @field_validator("priority") def _validate_priority(cls, v: str) -> str: if v is not None and v not in set(Priority):