Auto-fix lint and format issues

Author: octavia-squidington-iii

airbyte_cdk/sources/declarative/decoders/composite_raw_decoder.py

@@ -147,6 +147,7 @@ def parse(
 _HEADER = str
 _HEADER_VALUE = str
 
+
 class CompositeRawDecoder(Decoder):
     """
     Decoder strategy to transform a requests.Response into a Generator[MutableMapping[str, Any], None, None]
@@ -163,7 +164,12 @@ class CompositeRawDecoder(Decoder):
     """
 
     @classmethod
-    def by_headers(cls, parsers: List[Tuple[Set[_HEADER], Set[_HEADER_VALUE], Parser]], stream_response: bool, fallback_parser: Parser) -> "CompositeRawDecoder":
+    def by_headers(
+        cls,
+        parsers: List[Tuple[Set[_HEADER], Set[_HEADER_VALUE], Parser]],
+        stream_response: bool,
+        fallback_parser: Parser,
+    ) -> "CompositeRawDecoder":
         parsers_by_header = {}
         for headers, header_values, parser in parsers:
             for header in headers:
@@ -174,7 +180,12 @@ def by_headers(cls, parsers: List[Tuple[Set[_HEADER], Set[_HEADER_VALUE], Parser
     def from_parser(cls, parser: Parser, stream_response: bool) -> "CompositeRawDecoder":
         return cls(parser, stream_response, {})
 
-    def __init__(self, parser: Parser, stream_response: bool = True, parsers_by_header: Optional[Dict[_HEADER, Dict[_HEADER_VALUE, Parser]]] = None) -> None:
+    def __init__(
+        self,
+        parser: Parser,
+        stream_response: bool = True,
+        parsers_by_header: Optional[Dict[_HEADER, Dict[_HEADER_VALUE, Parser]]] = None,
+    ) -> None:
         self._parsers_by_header = parsers_by_header if parsers_by_header else {}
         self._fallback_parser = parser
         self._stream_response = stream_response
@@ -202,6 +213,9 @@ def decode(
 
     def _select_parser(self, response: requests.Response) -> Parser:
         for header, parser_by_header_value in self._parsers_by_header.items():
-            if header in response.headers and response.headers[header] in parser_by_header_value.keys():
+            if (
+                header in response.headers
+                and response.headers[header] in parser_by_header_value.keys()
+            ):
                 return parser_by_header_value[response.headers[header]]
         return self._fallback_parser

airbyte_cdk/sources/declarative/parsers/model_to_component_factory.py

@@ -2217,9 +2217,7 @@ def create_gzip_decoder(
     ) -> Decoder:
         gzip_parser: GzipParser = ModelToComponentFactory._get_parser(model, config)  # type: ignore  # based on the model, we know this will be a GzipParser
         return CompositeRawDecoder.by_headers(
-            [
-                ({"Content-Encoding", "Content-Type"}, _COMPRESSED_RESPONSE_TYPES, gzip_parser)
-            ],
+            [({"Content-Encoding", "Content-Type"}, _COMPRESSED_RESPONSE_TYPES, gzip_parser)],
             stream_response=False if self._emit_connector_builder_messages else True,
             fallback_parser=gzip_parser.inner_parser,
         )

unit_tests/sources/declarative/decoders/test_composite_decoder.py

@@ -134,7 +134,11 @@ def test_given_header_match_when_decode_then_select_parser(requests_mock):
 
     parser = GzipParser(inner_parser=JsonLineParser())
     unused_parser = Mock()
-    composite_raw_decoder = CompositeRawDecoder.by_headers([({"Content-Encoding"}, {"gzip"}, parser)], stream_response=True, fallback_parser=unused_parser)
+    composite_raw_decoder = CompositeRawDecoder.by_headers(
+        [({"Content-Encoding"}, {"gzip"}, parser)],
+        stream_response=True,
+        fallback_parser=unused_parser,
+    )
     counter = 0
     for _ in composite_raw_decoder.decode(response):
         counter += 1
@@ -151,7 +155,11 @@ def test_given_header_does_not_match_when_decode_then_select_fallback_parser(req
     response = requests.get("https://airbyte.io/", stream=True)
 
     unused_parser = GzipParser(inner_parser=Mock())
-    composite_raw_decoder = CompositeRawDecoder.by_headers([({"Content-Encoding"}, {"gzip"}, unused_parser)], stream_response=True, fallback_parser=JsonLineParser())
+    composite_raw_decoder = CompositeRawDecoder.by_headers(
+        [({"Content-Encoding"}, {"gzip"}, unused_parser)],
+        stream_response=True,
+        fallback_parser=JsonLineParser(),
+    )
     counter = 0
     for _ in composite_raw_decoder.decode(response):
         counter += 1