Skip to content

Commit 4f84c14

Browse files
committed
revert unnecessary changes
1 parent b8407e5 commit 4f84c14

File tree

2 files changed

+3
-10
lines changed

2 files changed

+3
-10
lines changed

airbyte_cdk/connector_builder/connector_builder_handler.py

Lines changed: 2 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -53,12 +53,7 @@ def get_limits(config: Mapping[str, Any]) -> TestLimits:
5353
max_slices = command_config.get(MAX_SLICES_KEY) or DEFAULT_MAXIMUM_NUMBER_OF_SLICES
5454
max_records = command_config.get(MAX_RECORDS_KEY) or DEFAULT_MAXIMUM_RECORDS
5555
max_streams = command_config.get(MAX_STREAMS_KEY) or DEFAULT_MAXIMUM_STREAMS
56-
return TestLimits(
57-
max_records=max_records,
58-
max_pages_per_slice=max_pages_per_slice,
59-
max_slices=max_slices,
60-
max_streams=max_streams,
61-
)
56+
return TestLimits(max_records, max_pages_per_slice, max_slices, max_streams)
6257

6358

6459
def create_source(config: Mapping[str, Any], limits: TestLimits) -> ManifestDeclarativeSource:
@@ -86,9 +81,7 @@ def read_stream(
8681
) -> AirbyteMessage:
8782
try:
8883
test_read_handler = TestReader(
89-
max_pages_per_slice=limits.max_pages_per_slice,
90-
max_slices=limits.max_slices,
91-
max_record_limit=limits.max_records,
84+
limits.max_pages_per_slice, limits.max_slices, limits.max_records
9285
)
9386
# The connector builder only supports a single stream
9487
stream_name = configured_catalog.streams[0].stream.name

unit_tests/connector_builder/test_connector_builder_handler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -979,7 +979,7 @@ def test_get_limits(
979979
assert limits.max_slices == expected_max_slices
980980

981981

982-
def test_create_source() -> None:
982+
def test_create_source():
983983
max_records = 3
984984
max_pages_per_slice = 2
985985
max_slices = 1

0 commit comments

Comments
 (0)