From d7b02798f4db27dade57fb12a4866acb7d7fbc87 Mon Sep 17 00:00:00 2001
From: epwalsh <petew@allenai.org>
Date: Wed, 29 May 2024 09:00:11 -0700
Subject: [PATCH] Add 2 new fields to `Experiment` spec

---
 CHANGELOG.md                    | 4 ++++
 beaker/data_model/experiment.py | 2 ++
 2 files changed, 6 insertions(+)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 10d6fdf..a09a9e7 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -9,6 +9,10 @@ use patch releases for compatibility fixes instead.
 
 ## Unreleased
 
+### Added
+
+- Added `Experiment.canceled` and `Experiment.canceled_for` fields.
+
 ## [v1.26.13](https://github.com/allenai/beaker-py/releases/tag/v1.26.13) - 2024-05-28
 
 ### Added
diff --git a/beaker/data_model/experiment.py b/beaker/data_model/experiment.py
index 464905c..f339b5f 100644
--- a/beaker/data_model/experiment.py
+++ b/beaker/data_model/experiment.py
@@ -18,6 +18,8 @@ class Experiment(BaseModel):
     description: Optional[str] = None
     author: Account
     created: datetime
+    canceled: Optional[datetime] = None
+    canceled_for: Optional[str] = None
     workspace_ref: WorkspaceRef
     jobs: Tuple[Job, ...] = Field(default_factory=tuple)