From 8bdbdeea5cde1cb85ba19f77ac74169b716ca0fd Mon Sep 17 00:00:00 2001
From: epwalsh <petew@allenai.org>
Date: Fri, 14 Jun 2024 09:31:17 -0700
Subject: [PATCH] Add `weka` option to `DataSource`

---
 CHANGELOG.md                         | 4 ++++
 beaker/data_model/experiment_spec.py | 5 +++++
 2 files changed, 9 insertions(+)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 52d6ef4..61a7828 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -9,6 +9,10 @@ use patch releases for compatibility fixes instead.
 
 ## Unreleased
 
+### Added
+
+- Added `weka` option to `DataSource` model.
+
 ### Fixed
 
 - Only check for upgrades once every 12 hours by default.
diff --git a/beaker/data_model/experiment_spec.py b/beaker/data_model/experiment_spec.py
index ef132f3..0dabce3 100644
--- a/beaker/data_model/experiment_spec.py
+++ b/beaker/data_model/experiment_spec.py
@@ -101,6 +101,11 @@ class DataSource(BaseModel, frozen=False):
     - ``/var/beaker/share`` as a shared local scratch space.
     """
 
+    weka: Optional[str] = None
+    """
+    The name of a weka bucket.
+    """
+
     result: Optional[str] = None
     """
     Name of a previous task whose result will be mounted.