Add two missing fields

allenai · Dec 11, 2024 · 2bb86d8 · 2bb86d8
1 parent e743373
commit 2bb86d8
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -9,6 +9,11 @@ use patch releases for compatibility fixes instead.
 
 ## Unreleased
 
+### Added
+
+- Added new field `Cluster.max_session_timeout: Optional[int]`.
+- Added new field `JobExecution.retry_ancestor: Optional[str]`.
+
 ### Fixed
 
 - Made checking for version upgrades more robust.

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

diff --git a/beaker/data_model/job.py b/beaker/data_model/job.py
@@ -133,6 +133,7 @@ class JobExecution(BaseModel):
     workspace: Optional[str] = None
     replica_rank: Optional[int] = None
     replica_group_id: Optional[str] = None
+    retry_ancestor: Optional[str] = None
 
 
 class JobKind(StrEnum):