Skip to content

Commit c6f1222

Browse files
committed
docs: add wait example for trainings to mirror v1
1 parent 734ec41 commit c6f1222

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

UPGRADING.md

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -442,7 +442,17 @@ training = replicate.trainings.create(
442442
)
443443

444444
# No instance methods available
445-
# Use resource methods
445+
# Use client methods instead
446+
447+
# Wait for training (no trainings.wait() available)
448+
# Poll with get() instead
449+
while True:
450+
training = replicate.trainings.get(training_id=training.id)
451+
if training.status in ["succeeded", "failed", "canceled"]:
452+
break
453+
time.sleep(1)
454+
455+
# Cancel training
446456
training = replicate.trainings.cancel(training_id=training.id)
447457
```
448458

0 commit comments

Comments
 (0)