From a700c50952ad41147e9fff8e9f657cc3d9468d19 Mon Sep 17 00:00:00 2001 From: dan Date: Wed, 1 Mar 2023 15:51:58 -0800 Subject: [PATCH] formatting --- replicate/version.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/replicate/version.py b/replicate/version.py index cf2574ce..27b7ae4d 100644 --- a/replicate/version.py +++ b/replicate/version.py @@ -6,6 +6,7 @@ from replicate.exceptions import ModelError from replicate.schema import make_schema_backwards_compatible + class Version(BaseModel): id: str created_at: datetime.datetime @@ -29,8 +30,8 @@ def predict(self, **kwargs) -> Union[Any, Iterator[Any]]: if prediction.status == "failed": raise ModelError(prediction.error) return prediction.output - - def create_prediction(self, **kwargs) -> 'Prediction': + + def create_prediction(self, **kwargs) -> "Prediction": """Creates and immediately returns a prediction""" return self._client.predictions.create(version=self, input=kwargs)