From 80d84b2d6ef55d345898237943a612ba809aa047 Mon Sep 17 00:00:00 2001 From: Mattt Zmuda Date: Wed, 12 Apr 2023 07:03:22 -0700 Subject: [PATCH 1/3] Relax Training version type constraint to be Optional Signed-off-by: Mattt Zmuda --- replicate/training.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/replicate/training.py b/replicate/training.py index 06391d2d..0202c492 100644 --- a/replicate/training.py +++ b/replicate/training.py @@ -6,6 +6,7 @@ from replicate.exceptions import ReplicateException from replicate.files import upload_file from replicate.json import encode_json +from replicate.version import Version class Training(BaseModel): @@ -19,7 +20,7 @@ class Training(BaseModel): output: Optional[Any] started_at: Optional[str] status: str - version: str + version: Optional[Version] def cancel(self) -> None: """Cancel a running training""" From c528ee46a4c77a40847a39462a079a55c5b53aa7 Mon Sep 17 00:00:00 2001 From: Mattt Zmuda Date: Wed, 12 Apr 2023 07:08:19 -0700 Subject: [PATCH 2/3] Delete version field from JSON response in trainnings.get Signed-off-by: Mattt Zmuda --- replicate/training.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/replicate/training.py b/replicate/training.py index 0202c492..058873e5 100644 --- a/replicate/training.py +++ b/replicate/training.py @@ -45,6 +45,8 @@ def get(self, id: str) -> Training: f"/v1/trainings/{id}", ) obj = resp.json() + # HACK: resolve this? make it lazy somehow? + del obj["version"] return self.prepare_model(obj) def create( # type: ignore From 77441555269419130fd0e6124ff6b35eb045b0d9 Mon Sep 17 00:00:00 2001 From: Mattt Zmuda Date: Wed, 12 Apr 2023 07:08:38 -0700 Subject: [PATCH 3/3] Remove version field from trainings.create response Signed-off-by: Mattt Zmuda --- replicate/training.py | 1 + 1 file changed, 1 insertion(+) diff --git a/replicate/training.py b/replicate/training.py index 058873e5..d7e97bc3 100644 --- a/replicate/training.py +++ b/replicate/training.py @@ -86,4 +86,5 @@ def create( # type: ignore json=body, ) obj = resp.json() + del obj["version"] return self.prepare_model(obj)