diff --git a/CHANGELOG.md b/CHANGELOG.md index 23bb456..8603173 100644 --- 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 index 37311e0..b2df33f 100644 --- 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.