Add some new missing fields (#276)

allenai · May 10, 2024 · 3abc2d1 · 3abc2d1
1 parent 994482d
commit 3abc2d1
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -12,6 +12,8 @@ use patch releases for compatibility fixes instead.
 ### Added
 
 - Added `preemptible` argument to `TaskSpec.new()`.
+- Added `Workspace.max_workload_priority` field.
+- Added `Job.preemptible` field.
 
 ## [v1.26.11](https://github.com/allenai/beaker-py/releases/tag/v1.26.11) - 2024-05-10
 

diff --git a/beaker/data_model/job.py b/beaker/data_model/job.py
@@ -182,6 +182,7 @@ class Job(BaseModel):
     host_networking: bool = False
     port_mappings: Optional[Dict[str, int]] = None
     result: Optional[ExecutionResult] = None
+    preemptible: Optional[bool] = None
 
     @property
     def display_name(self) -> str:

diff --git a/beaker/data_model/workspace.py b/beaker/data_model/workspace.py
@@ -3,6 +3,7 @@
 
 from .account import Account
 from .base import BaseModel, BasePage, StrEnum
+from .experiment_spec import Priority
 
 __all__ = [
     "WorkspaceSize",
@@ -38,6 +39,7 @@ class Workspace(BaseModel):
     created: datetime
     modified: datetime
     archived: bool = False
+    max_workload_priority: Optional[Priority] = None
 
 
 class WorkspaceRef(BaseModel):