format and lint

Author: philandstuff

replicate/client.py

@@ -351,7 +351,7 @@ def close(self) -> None:
 def _get_api_token_from_environment() -> Optional[str]:
     """Get API token from cog current scope if available, otherwise from environment."""
     try:
-        import cog
+        import cog  # noqa: I001 # pyright: ignore [reportMissingImports]
 
         for key, value in cog.current_scope().context.items():
             if key.upper() == "REPLICATE_API_TOKEN":

tests/test_client.py

@@ -98,9 +98,9 @@ def test_custom_headers_are_applied():
 
     def mock_send(request):
         assert "User-Agent" in request.headers, "Custom header not found in request"
-        assert (
-            request.headers["User-Agent"] == "my-custom-user-agent/1.0"
-        ), "Custom header value is incorrect"
+        assert request.headers["User-Agent"] == "my-custom-user-agent/1.0", (
+            "Custom header value is incorrect"
+        )
         return httpx.Response(401, json={})
 
     mock_send_wrapper = mock.Mock(side_effect=mock_send)