style: fix formatting and linting issues

Co-Authored-By: [email protected] <[email protected]>
airbytehq · Feb 7, 2025 · e1907ff · e1907ff
1 parent 9b439fd
commit e1907ff
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 7 deletions.
diff --git a/airbyte_cdk/sources/declarative/decoders/csv_decoder.py b/airbyte_cdk/sources/declarative/decoders/csv_decoder.py
@@ -36,7 +36,7 @@ def decode(
                 return
 
             # First validate CSV structure
-            lines = response.text.strip().split('\n')
+            lines = response.text.strip().split("\n")
             if not lines:
                 yield {}
                 return
@@ -55,7 +55,7 @@ def decode(
                     iterator=True,
                     dtype=object,
                     delimiter=self.delimiter,
-                    encoding=self.encoding
+                    encoding=self.encoding,
                 )
             except (pd.errors.EmptyDataError, pd.errors.ParserError):
                 yield {}

diff --git a/airbyte_cdk/sources/declarative/parsers/model_to_component_factory.py b/airbyte_cdk/sources/declarative/parsers/model_to_component_factory.py
@@ -59,14 +59,14 @@
 from airbyte_cdk.sources.declarative.datetime import MinMaxDatetime
 from airbyte_cdk.sources.declarative.declarative_stream import DeclarativeStream
 from airbyte_cdk.sources.declarative.decoders import (
+    CsvDecoder,
     Decoder,
     GzipJsonDecoder,
     IterableDecoder,
     JsonDecoder,
     JsonlDecoder,
     PaginationDecoderDecorator,
     XmlDecoder,
-    CsvDecoder,
     ZipfileDecoder,
 )
 from airbyte_cdk.sources.declarative.decoders.composite_raw_decoder import (
@@ -155,6 +155,9 @@
 from airbyte_cdk.sources.declarative.models.declarative_component_schema import (
     ConstantBackoffStrategy as ConstantBackoffStrategyModel,
 )
+from airbyte_cdk.sources.declarative.models.declarative_component_schema import (
+    CsvDecoder as CsvDecoderModel,
+)
 from airbyte_cdk.sources.declarative.models.declarative_component_schema import (
     CsvParser as CsvParserModel,
 )
@@ -367,9 +370,6 @@
 from airbyte_cdk.sources.declarative.models.declarative_component_schema import (
     XmlDecoder as XmlDecoderModel,
 )
-from airbyte_cdk.sources.declarative.models.declarative_component_schema import (
-    CsvDecoder as CsvDecoderModel,
-)
 from airbyte_cdk.sources.declarative.models.declarative_component_schema import (
     ZipfileDecoder as ZipfileDecoderModel,
 )

diff --git a/unit_tests/sources/declarative/decoders/test_csv_decoder.py b/unit_tests/sources/declarative/decoders/test_csv_decoder.py
@@ -15,7 +15,11 @@
         ("name;age\nJohn;30", [{"name": "John", "age": "30"}], ";"),
         ("", [{}], ","),  # Empty response
         ("invalid,csv,data\nno,columns", [{}], ","),  # Malformed CSV
-        ("name,age\nJohn,30\nJane,25", [{"name": "John", "age": "30"}, {"name": "Jane", "age": "25"}], ","),  # Multiple rows
+        (
+            "name,age\nJohn,30\nJane,25",
+            [{"name": "John", "age": "30"}, {"name": "Jane", "age": "25"}],
+            ",",
+        ),  # Multiple rows
     ],
 )
 def test_csv_decoder(requests_mock, response_body, expected, delimiter):