chore: formatting

Author: dgellow

README.md

@@ -40,7 +40,9 @@ seedream = replicate.use("bytedance/seedream-4")
 veo = replicate.use("google/veo-3-fast")
 
 # Enhance a simple prompt
-image_prompt = claude(prompt="bananas wearing cowboy hats", system_prompt="turn prompts into image prompts")
+image_prompt = claude(
+    prompt="bananas wearing cowboy hats", system_prompt="turn prompts into image prompts"
+)
 
 # Generate an image from the enhanced prompt
 images = seedream(prompt=image_prompt)
@@ -59,9 +61,7 @@ The library uses the `REPLICATE_API_TOKEN` environment variable by default to im
 import os
 from replicate import Replicate
 
-client = Replicate(
-    bearer_token=os.environ.get("REPLICATE_API_TOKEN")
-)
+client = Replicate(bearer_token=os.environ.get("REPLICATE_API_TOKEN"))
 ```
 
 ## Using `replicate.use()`

tests/lib/test_stream.py

@@ -101,8 +101,9 @@ def test_stream_returns_iterator():
 
 def test_stream_works_same_as_use_with_streaming():
     """Test that stream() produces the same output as use() with streaming=True."""
-    with patch("replicate.lib._stream.use") as mock_stream_use, \
-         patch("replicate.lib._predictions_use.use") as mock_predictions_use:
+    with patch("replicate.lib._stream.use") as mock_stream_use, patch(
+        "replicate.lib._predictions_use.use"
+    ) as mock_predictions_use:
         # Create a mock function that returns an iterator
         mock_function = Mock()
         expected_output = ["Test", " ", "output"]