Set version field in trainings.create

mattt · mattt · commit ea3bf4e1c42f · 2023-04-12T07:08:44.000-07:00
Signed-off-by: Mattt Zmuda &lt;mattt@replicate.com&gt;
diff --git a/replicate/training.py b/replicate/training.py
@@ -86,4 +86,5 @@ def create(  # type: ignore
             json=body,
         )
         obj = resp.json()
+        obj["version"] = version
         return self.prepare_model(obj)

Original file line number	Diff line number	Diff line change
`@@ -86,4 +86,5 @@ def create( # type: ignore`
`86`	`86`	`json=body,`
`87`	`87`	`)`
`88`	`88`	`obj = resp.json()`
	`89`	`+ obj["version"] = version`
`89`	`90`	`return self.prepare_model(obj)`