From f25a8651e01c21fe1e0070336673154c74756078 Mon Sep 17 00:00:00 2001
From: Pete <epwalsh10@gmail.com>
Date: Fri, 10 May 2024 16:17:00 -0700
Subject: [PATCH] Add field `TaskContext.preemptible` (#275)

---
 CHANGELOG.md                         | 4 ++++
 beaker/data_model/experiment_spec.py | 5 +++++
 2 files changed, 9 insertions(+)

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):