diff --git a/replicate/prediction.py b/replicate/prediction.py index b4ff047..6e373ef 100644 --- a/replicate/prediction.py +++ b/replicate/prediction.py @@ -55,7 +55,7 @@ class Prediction(Resource): version: str """An identifier for the version of the model used to create the prediction.""" - status: Literal["starting", "processing", "succeeded", "failed", "canceled"] + status: Literal["starting", "processing", "succeeded", "failed", "canceled", "aborted"] """The status of the prediction.""" input: Optional[Dict[str, Any]]