Sourcery refactored main branch

replicate/prediction.py

@@ -69,8 +69,7 @@ def output_iterator(self) -> Iterator[Any]:
         previous_output = self.output or []
         while self.status not in ["succeeded", "failed", "canceled"]:
             output = self.output or []
-            new_output = output[len(previous_output) :]
-            yield from new_output
+            yield from output[len(previous_output) :]
             previous_output = output
             time.sleep(self._client.poll_interval)
             self.reload()
@@ -79,9 +78,7 @@ def output_iterator(self) -> Iterator[Any]:
             raise ModelError(self.error)
 
         output = self.output or []
-        new_output = output[len(previous_output) :]
-        for output in new_output:
-            yield output
+        yield from output[len(previous_output) :]
 
     def cancel(self) -> None:
         """

replicate/version.py

@@ -65,8 +65,7 @@ def predict(self, **kwargs) -> Union[Any, Iterator[Any]]:
 
     def get_transformed_schema(self) -> dict:
         schema = self.openapi_schema
-        schema = make_schema_backwards_compatible(schema, self.cog_version)
-        return schema
+        return make_schema_backwards_compatible(schema, self.cog_version)
 
 
 class VersionCollection(Collection):

tests/factories.py

@@ -8,8 +8,7 @@
 
 
 def create_client():
-    client = Client(api_token="abc123")
-    return client
+    return Client(api_token="abc123")
 
 
 def get_mock_schema():