diff --git a/replicate/training.py b/replicate/training.py
index 06391d2d..d7e97bc3 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"""
@@ -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)