allow numbers in exp spec cmd field (#269)

allenai · Apr 18, 2024 · 289cae6 · 289cae6
1 parent 5cc5986
commit 289cae6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -9,6 +9,10 @@ use patch releases for compatibility fixes instead.
 
 ## Unreleased
 
+### Fixed
+
+- Numbers are now allowed in `command` and `arguments` fields for experiment specs.
+
 ## [v1.26.4](https://github.com/allenai/beaker-py/releases/tag/v1.26.4) - 2024-04-10
 
 ### Fixed

diff --git a/beaker/data_model/experiment_spec.py b/beaker/data_model/experiment_spec.py
@@ -348,7 +348,7 @@ class TaskSpec(BaseModel, frozen=False):
     It must be unique among all tasks within its experiment.
     """
 
-    command: Optional[List[str]] = None
+    command: Optional[List[Union[str, int, float]]] = None
     """
     Command is the full shell command to run as a sequence of separate arguments.
 
@@ -358,7 +358,7 @@ class TaskSpec(BaseModel, frozen=False):
     Example: ``["python", "-u", "main.py"]``
     """
 
-    arguments: Optional[List[str]] = None
+    arguments: Optional[List[Union[str, int, float]]] = None
     """
     Arguments are appended to the command and replace default arguments such as Docker's ``CMD`` directive.