relax Docker layer upload/download status data model (#291)

allenai · Nov 14, 2024 · e8eb8c7 · e8eb8c7
1 parent c58c5e9
commit e8eb8c7
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
+
+- Relaxed requirements on Docker layer upload/download status strings.
+
 ## [v1.32.1](https://github.com/allenai/beaker-py/releases/tag/v1.32.1) - 2024-11-01
 
 ### Added

diff --git a/beaker/data_model/image.py b/beaker/data_model/image.py
@@ -90,7 +90,7 @@ class DockerLayerDownloadStatus(StrEnum):
 
 class DockerLayerUploadState(BaseModel):
     id: str
-    status: DockerLayerUploadStatus
+    status: str
     progress_detail: DockerLayerProgress
     progress: Optional[str] = None
 
@@ -101,7 +101,7 @@ def _validate_status(cls, v: str) -> str:
 
 class DockerLayerDownloadState(BaseModel):
     id: str
-    status: DockerLayerDownloadStatus
+    status: str
     progress_detail: DockerLayerProgress
     progress: Optional[str] = None