Add Cluster.require_preemptible_tasks field

allenai · May 28, 2024 · 32c9a73 · 32c9a73
1 parent 4393e03
commit 32c9a73
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -12,6 +12,7 @@ use patch releases for compatibility fixes instead.
 ### Added
 
 - Added `Secret.author_id` field.
+- Added `Cluster.require_preemptible_tasks` field.
 
 ## [v1.26.12](https://github.com/allenai/beaker-py/releases/tag/v1.26.12) - 2024-05-10
 

diff --git a/beaker/data_model/cluster.py b/beaker/data_model/cluster.py
@@ -42,6 +42,7 @@ class Cluster(BaseModel):
     allow_preemptible_restriction_exceptions: Optional[bool] = None
     compute_source: Optional[str] = None
     max_job_timeout: Optional[int] = None
+    require_preemptible_tasks: Optional[bool] = None
 
     @field_validator("validated")
     def _validate_datetime(cls, v: Optional[datetime]) -> Optional[datetime]: