diff --git a/CHANGELOG.md b/CHANGELOG.md index 8cb09c9..88b06d4 100644 --- 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 index 0a1689a..833152a 100644 --- 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 index 8b13569..3303c4c 100644 --- 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):