Add Job.node_has_gpus field

allenai · Jul 25, 2024 · f5f5a25 · f5f5a25
1 parent 48eb2d6
commit f5f5a25
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -9,6 +9,10 @@ use patch releases for compatibility fixes instead.
 
 ## Unreleased
 
+### Added
+
+- Added `node_has_gpus` field to `Job`.
+
 ## [v1.31.1](https://github.com/allenai/beaker-py/releases/tag/v1.31.1) - 2024-07-14
 
 ### Fixed

diff --git a/beaker/data_model/job.py b/beaker/data_model/job.py
@@ -177,6 +177,7 @@ class Job(BaseModel):
     execution: Optional[JobExecution] = None
     execution_results: Optional[Dict[str, Any]] = None
     node: Optional[str] = None
+    node_has_gpus: Optional[bool] = None
     requests: Optional[JobRequests] = None
     limits: Optional[JobLimits] = None
     session: Optional[Session] = None