Update deprecation warnings for Model.predict and Version.predict (#137)

* Add deprecation warning to docstring for Version.predict

Signed-off-by: Mattt Zmuda <[email protected]>

* Update deprecation message to use replicate.run instead of version.predict

Signed-off-by: Mattt Zmuda <[email protected]>

---------

Signed-off-by: Mattt Zmuda <[email protected]>

Author: mattt

replicate/model.py

@@ -23,11 +23,11 @@ class Model(BaseModel):
 
     def predict(self, *args, **kwargs) -> None:
         """
-        DEPRECATED: Use `version.predict()` instead.
+        DEPRECATED: Use `replicate.run()` instead.
         """
 
         raise ReplicateException(
-            "The `model.predict()` method has been removed, because it's unstable: if a new version of the model you're using is pushed and its API has changed, your code may break. Use `version.predict()` instead. See https://github.com/replicate/replicate-python#readme"
+            "The `model.predict()` method has been removed, because it's unstable: if a new version of the model you're using is pushed and its API has changed, your code may break. Use `replicate.run()` instead. See https://github.com/replicate/replicate-python#readme"
         )
 
     @property

replicate/version.py

@@ -32,6 +32,8 @@ class Version(BaseModel):
 
     def predict(self, **kwargs) -> Union[Any, Iterator[Any]]:
         """
+        DEPRECATED: Use `replicate.run()` instead.
+
         Create a prediction using this model version.
 
         Args: