Fix warnings

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

Author: mattt

replicate/helpers.py

@@ -117,7 +117,7 @@ class FileOutput(httpx.SyncByteStream, httpx.AsyncByteStream):
     url: str
     client: "Client"
 
-    def __init__(self, url: str, client: "Client"):
+    def __init__(self, url: str, client: "Client") -> None:
         self.url = url
         self.client = client
 
@@ -129,8 +129,7 @@ def read(self) -> bytes:
     def __iter__(self) -> Iterator[bytes]:
         with self.client._client.stream("GET", self.url) as response:
             response.raise_for_status()
-            for chunk in response.iter_bytes():
-                yield chunk
+            yield from response.iter_bytes()
 
     async def aread(self) -> bytes:
         async with self.client._async_client.stream("GET", self.url) as response:

replicate/model.py

@@ -3,8 +3,8 @@
 from typing_extensions import NotRequired, TypedDict, Unpack, deprecated
 
 from replicate.exceptions import ReplicateException
-from replicate.identifier import ModelVersionIdentifier
 from replicate.helpers import async_encode_json, encode_json
+from replicate.identifier import ModelVersionIdentifier
 from replicate.pagination import Page
 from replicate.prediction import (
     Prediction,

replicate/training.py

@@ -13,8 +13,8 @@
 
 from typing_extensions import NotRequired, Unpack
 
-from replicate.identifier import ModelVersionIdentifier
 from replicate.helpers import async_encode_json, encode_json
+from replicate.identifier import ModelVersionIdentifier
 from replicate.model import Model
 from replicate.pagination import Page
 from replicate.resource import Namespace, Resource