Skip to content

Commit 9f1b532

Browse files
committed
run ruff format
1 parent 0c784b3 commit 9f1b532

File tree

3 files changed

+18
-5
lines changed

3 files changed

+18
-5
lines changed

airbyte_cdk/connector_builder/connector_builder_handler.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -108,7 +108,12 @@ def read_stream(
108108
stream_name = configured_catalog.streams[0].stream.name
109109

110110
stream_read = test_read_handler.run_test_read(
111-
source, config, configured_catalog, stream_name, state, limits.max_records,
111+
source,
112+
config,
113+
configured_catalog,
114+
stream_name,
115+
state,
116+
limits.max_records,
112117
)
113118

114119
return AirbyteMessage(

airbyte_cdk/connector_builder/test_reader/helpers.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -268,7 +268,10 @@ def should_close_page_for_slice(at_least_one_page_in_group: bool, message: Airby
268268
"""
269269
return at_least_one_page_in_group and should_process_slice_descriptor(message)
270270

271-
def is_page_http_request_for_different_stream(json_message: Optional[Dict[str, Any]], stream_name: str) -> bool:
271+
272+
def is_page_http_request_for_different_stream(
273+
json_message: Optional[Dict[str, Any]], stream_name: str
274+
) -> bool:
272275
"""
273276
Determines whether a given JSON message represents a page HTTP request for a different stream.
274277
@@ -285,8 +288,11 @@ def is_page_http_request_for_different_stream(json_message: Optional[Dict[str, A
285288
Returns:
286289
bool: True if the JSON message is a page HTTP request for a different stream, False otherwise.
287290
"""
288-
return is_page_http_request(json_message) and json_message.get("airbyte_cdk", {}).get("stream", {}).get("name", {}) != stream_name
289-
291+
return (
292+
is_page_http_request(json_message)
293+
and json_message.get("airbyte_cdk", {}).get("stream", {}).get("name", {}) != stream_name
294+
)
295+
290296

291297
def is_page_http_request(json_message: Optional[Dict[str, Any]]) -> bool:
292298
"""

airbyte_cdk/connector_builder/test_reader/reader.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -121,7 +121,9 @@ def run_test_read(
121121

122122
schema_inferrer = SchemaInferrer(
123123
self._pk_to_nested_and_composite_field(stream.primary_key) if stream else None,
124-
self._cursor_field_to_nested_and_composite_field(stream.cursor_field) if stream else None,
124+
self._cursor_field_to_nested_and_composite_field(stream.cursor_field)
125+
if stream
126+
else None,
125127
)
126128
datetime_format_inferrer = DatetimeFormatInferrer()
127129

0 commit comments

Comments
 (0)