lint

Author: zeke

src/replicate/_client.py

@@ -4,25 +4,25 @@
 
 import os
 from typing import (
+    TYPE_CHECKING,
     Any,
-    AsyncIterator,
-    Callable,
-    Iterator,
+    Union,
     Literal,
     Mapping,
-    Optional,
-    TYPE_CHECKING,
     TypeVar,
-    Union,
+    Callable,
+    Iterator,
+    Optional,
+    AsyncIterator,
     cast,
     overload,
 )
-from typing_extensions import ParamSpec, Self, Unpack, override
+from typing_extensions import Self, Unpack, ParamSpec, override
 
 import httpx
 
-from replicate._utils._logs import logger
 from replicate.lib._files import FileEncodingStrategy
+from replicate._utils._logs import logger
 from replicate.lib._predictions_run import Model, Version, ModelVersionIdentifier
 from replicate.types.prediction_create_params import PredictionCreateParamsWithoutVersion
 
@@ -82,28 +82,28 @@ def _get_api_token_from_environment() -> str | None:
     """Get API token from cog current scope if available, otherwise from environment."""
     try:
         import cog  # type: ignore[import-untyped, import-not-found]
-        
+
         # Get the current scope - this might return None or raise an exception
         scope = getattr(cog, "current_scope", lambda: None)()
         if scope is None:
             return os.environ.get("REPLICATE_API_TOKEN")
-            
+
         # Get the context from the scope
         context = getattr(scope, "context", None)
         if context is None:
             return os.environ.get("REPLICATE_API_TOKEN")
-            
+
         # Get the items method and call it
         items_method = getattr(context, "items", None)
         if not callable(items_method):
             return os.environ.get("REPLICATE_API_TOKEN")
-            
+
         # Iterate through context items looking for the API token
         items = cast(Iterator[tuple[Any, Any]], items_method())
         for key, value in items:
             if str(key).upper() == "REPLICATE_API_TOKEN":
                 return str(value) if value is not None else value
-                
+
     except Exception as e:  # Catch all exceptions to ensure robust fallback
         logger.debug("Failed to retrieve API token from cog.current_scope(): %s", e)
 

src/replicate/types/prediction_create_params.py

@@ -35,7 +35,7 @@ class PredictionCreateParamsWithoutVersion(TypedDict, total=False):
     - you don't want to upload and host the file somewhere
     - you don't need to use the file again (Replicate will not store it)
     """
-    
+
     stream: bool
     """**This field is deprecated.**
 

tests/test_current_scope.py

@@ -149,7 +149,7 @@ def test_sync_client_uses_current_scope_token(self):
         mock_scope.context = {"REPLICATE_API_TOKEN": "cog-token"}
         mock_cog = mock.MagicMock()
         mock_cog.current_scope.return_value = mock_scope
-        
+
         # Clear environment variable to ensure we're using cog
         with mock.patch.dict(os.environ, {}, clear=True):
             with mock.patch.dict(sys.modules, {"cog": mock_cog}):
@@ -162,7 +162,7 @@ def test_async_client_uses_current_scope_token(self):
         mock_scope.context = {"REPLICATE_API_TOKEN": "cog-token"}
         mock_cog = mock.MagicMock()
         mock_cog.current_scope.return_value = mock_scope
-        
+
         # Clear environment variable to ensure we're using cog
         with mock.patch.dict(os.environ, {}, clear=True):
             with mock.patch.dict(sys.modules, {"cog": mock_cog}):
@@ -203,7 +203,7 @@ def test_explicit_token_overrides_current_scope(self):
         mock_scope.context = {"REPLICATE_API_TOKEN": "cog-token"}
         mock_cog = mock.MagicMock()
         mock_cog.current_scope.return_value = mock_scope
-        
+
         with mock.patch.dict(sys.modules, {"cog": mock_cog}):
             client = Replicate(bearer_token="explicit-token", base_url="http://test.example.com")
             assert client.bearer_token == "explicit-token"
@@ -214,7 +214,7 @@ def test_explicit_async_token_overrides_current_scope(self):
         mock_scope.context = {"REPLICATE_API_TOKEN": "cog-token"}
         mock_cog = mock.MagicMock()
         mock_cog.current_scope.return_value = mock_scope
-        
+
         with mock.patch.dict(sys.modules, {"cog": mock_cog}):
             client = AsyncReplicate(bearer_token="explicit-token", base_url="http://test.example.com")
-            assert client.bearer_token == "explicit-token"
+            assert client.bearer_token == "explicit-token"