Skip to content

Commit e1907ff

Browse files
style: fix formatting and linting issues
Co-Authored-By: [email protected] <[email protected]>
1 parent 9b439fd commit e1907ff

File tree

3 files changed

+11
-7
lines changed

3 files changed

+11
-7
lines changed

airbyte_cdk/sources/declarative/decoders/csv_decoder.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ def decode(
3636
return
3737

3838
# First validate CSV structure
39-
lines = response.text.strip().split('\n')
39+
lines = response.text.strip().split("\n")
4040
if not lines:
4141
yield {}
4242
return
@@ -55,7 +55,7 @@ def decode(
5555
iterator=True,
5656
dtype=object,
5757
delimiter=self.delimiter,
58-
encoding=self.encoding
58+
encoding=self.encoding,
5959
)
6060
except (pd.errors.EmptyDataError, pd.errors.ParserError):
6161
yield {}

airbyte_cdk/sources/declarative/parsers/model_to_component_factory.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -59,14 +59,14 @@
5959
from airbyte_cdk.sources.declarative.datetime import MinMaxDatetime
6060
from airbyte_cdk.sources.declarative.declarative_stream import DeclarativeStream
6161
from airbyte_cdk.sources.declarative.decoders import (
62+
CsvDecoder,
6263
Decoder,
6364
GzipJsonDecoder,
6465
IterableDecoder,
6566
JsonDecoder,
6667
JsonlDecoder,
6768
PaginationDecoderDecorator,
6869
XmlDecoder,
69-
CsvDecoder,
7070
ZipfileDecoder,
7171
)
7272
from airbyte_cdk.sources.declarative.decoders.composite_raw_decoder import (
@@ -155,6 +155,9 @@
155155
from airbyte_cdk.sources.declarative.models.declarative_component_schema import (
156156
ConstantBackoffStrategy as ConstantBackoffStrategyModel,
157157
)
158+
from airbyte_cdk.sources.declarative.models.declarative_component_schema import (
159+
CsvDecoder as CsvDecoderModel,
160+
)
158161
from airbyte_cdk.sources.declarative.models.declarative_component_schema import (
159162
CsvParser as CsvParserModel,
160163
)
@@ -367,9 +370,6 @@
367370
from airbyte_cdk.sources.declarative.models.declarative_component_schema import (
368371
XmlDecoder as XmlDecoderModel,
369372
)
370-
from airbyte_cdk.sources.declarative.models.declarative_component_schema import (
371-
CsvDecoder as CsvDecoderModel,
372-
)
373373
from airbyte_cdk.sources.declarative.models.declarative_component_schema import (
374374
ZipfileDecoder as ZipfileDecoderModel,
375375
)

unit_tests/sources/declarative/decoders/test_csv_decoder.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,11 @@
1515
("name;age\nJohn;30", [{"name": "John", "age": "30"}], ";"),
1616
("", [{}], ","), # Empty response
1717
("invalid,csv,data\nno,columns", [{}], ","), # Malformed CSV
18-
("name,age\nJohn,30\nJane,25", [{"name": "John", "age": "30"}, {"name": "Jane", "age": "25"}], ","), # Multiple rows
18+
(
19+
"name,age\nJohn,30\nJane,25",
20+
[{"name": "John", "age": "30"}, {"name": "Jane", "age": "25"}],
21+
",",
22+
), # Multiple rows
1923
],
2024
)
2125
def test_csv_decoder(requests_mock, response_body, expected, delimiter):

0 commit comments

Comments
 (0)