Fix 'B028 No explicit stacklevel keyword argument found'

mattt · mattt · commit 8dce26b6c02f · 2023-04-10T04:33:06.000-07:00
diff --git a/replicate/version.py b/replicate/version.py
@@ -18,6 +18,7 @@ def predict(self, **kwargs) -> Union[Any, Iterator[Any]]:
         warnings.warn(
             "version.predict() is deprecated. Use replicate.run() instead. It will be removed before version 1.0.",
             DeprecationWarning,
+            stacklevel=1,
         )
 
         prediction = self._client.predictions.create(version=self, input=kwargs)

Original file line number	Diff line number	Diff line change
`@@ -18,6 +18,7 @@ def predict(self, **kwargs) -> Union[Any, Iterator[Any]]:`
`18`	`18`	`warnings.warn(`
`19`	`19`	`"version.predict() is deprecated. Use replicate.run() instead. It will be removed before version 1.0.",`
`20`	`20`	`DeprecationWarning,`
	`21`	`+ stacklevel=1,`
`21`	`22`	`)`
`22`	`23`
`23`	`24`	`prediction = self._client.predictions.create(version=self, input=kwargs)`