style: lint

Author: anna-singleton-resolver

src/resolver_athena_client/client/athena_client.py

@@ -177,7 +177,9 @@ async def classify_single(
 
         # Apply compression if enabled
         if self.options.compress_images:
-            processed_image = compress_image(processed_image, self.options.compression_quality)
+            processed_image = compress_image(
+                processed_image, self.options.compression_quality
+            )
 
         request_encoding = (
             RequestEncoding.REQUEST_ENCODING_BROTLI
@@ -243,7 +245,9 @@ async def transform_image(image_data: ImageData) -> ClassificationInput:
 
             # Apply compression if enabled
             if self.options.compress_images:
-                compressed_image = compress_image(resized_image, self.options.compression_quality)
+                compressed_image = compress_image(
+                    resized_image, self.options.compression_quality
+                )
             else:
                 compressed_image = resized_image
 
@@ -254,7 +258,6 @@ async def transform_image(image_data: ImageData) -> ClassificationInput:
                 else RequestEncoding.REQUEST_ENCODING_UNCOMPRESSED
             )
 
-
             # Ensure we never send UNSPECIFIED format over the API
             image_format = compressed_image.image_format
             if image_format == ImageFormat.IMAGE_FORMAT_UNSPECIFIED:

src/resolver_athena_client/client/models/input_model.py

@@ -67,7 +67,9 @@ async def image_stream():
 
     """
 
-    def __init__(self, image_bytes: bytes, correlation_id: None | str = None) -> None:
+    def __init__(
+        self, image_bytes: bytes, correlation_id: None | str = None
+    ) -> None:
         """Initialize ImageData with bytes and calculate hashes.
 
         Args:

tests/test_classify_single.py

@@ -122,7 +122,9 @@ async def test_classify_single_with_correlation_id(
         return_value=mock_output
     )
 
-    copied_image_data = ImageData(sample_image_data.data, correlation_id=custom_correlation_id)
+    copied_image_data = ImageData(
+        sample_image_data.data, correlation_id=custom_correlation_id
+    )
 
     # Call classify_single with custom correlation ID
     _ = await athena_client.classify_single(copied_image_data)