Fix Pydantic validation errors for Training responses (#90)

* Relax Training version type constraint to be Optional

Signed-off-by: Mattt Zmuda <[email protected]>

* Delete version field from JSON response in trainnings.get

Signed-off-by: Mattt Zmuda <[email protected]>

* Remove version field from trainings.create response

Signed-off-by: Mattt Zmuda <[email protected]>

---------

Signed-off-by: Mattt Zmuda <[email protected]>

Author: mattt

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"""
@@ -44,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
@@ -83,4 +86,5 @@ def create(  # type: ignore
             json=body,
         )
         obj = resp.json()
+        del obj["version"]
         return self.prepare_model(obj)