Add weka option to DataSource

allenai · Jun 14, 2024 · 8bdbdee · 8bdbdee
1 parent df27ac8
commit 8bdbdee
Show file tree

Hide file tree

Showing 2 changed files with 9 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 `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
@@ -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.