Skip to content

Commit 7bec79b

Browse files
committed
Merge branch 'lazebnyi/add-cocurency-to-partial-declarative-sources' of github.com:airbytehq/airbyte-python-cdk into lazebnyi/add-cocurency-to-partial-declarative-sources
2 parents 90002a8 + 4a105a6 commit 7bec79b

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

airbyte_cdk/sources/declarative/concurrent_declarative_source.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,13 +46,13 @@
4646
from airbyte_cdk.sources.source import TState
4747
from airbyte_cdk.sources.streams import Stream
4848
from airbyte_cdk.sources.streams.concurrent.abstract_stream import AbstractStream
49+
from airbyte_cdk.sources.streams.concurrent.abstract_stream_facade import AbstractStreamFacade
4950
from airbyte_cdk.sources.streams.concurrent.availability_strategy import (
5051
AlwaysAvailableAvailabilityStrategy,
5152
)
5253
from airbyte_cdk.sources.streams.concurrent.cursor import ConcurrentCursor, FinalStateCursor
5354
from airbyte_cdk.sources.streams.concurrent.default_stream import DefaultStream
5455
from airbyte_cdk.sources.streams.concurrent.helpers import get_primary_key_from_stream
55-
from airbyte_cdk.sources.streams.concurrent.abstract_stream_facade import AbstractStreamFacade
5656

5757

5858
class ConcurrentDeclarativeSource(ManifestDeclarativeSource, Generic[TState]):
@@ -381,7 +381,10 @@ def _group_streams(
381381
# TODO: Remove this. This check is necessary to safely migrate Stripe during the transition state.
382382
# Condition below needs to ensure that concurrent support is not lost for sources that already support
383383
# it before migration, but now are only partially migrated to declarative implementation (e.g., Stripe).
384-
elif isinstance(declarative_stream, AbstractStreamFacade) and self.is_partially_declarative:
384+
elif (
385+
isinstance(declarative_stream, AbstractStreamFacade)
386+
and self.is_partially_declarative
387+
):
385388
concurrent_streams.append(declarative_stream.get_underlying_stream())
386389
else:
387390
synchronous_streams.append(declarative_stream)

0 commit comments

Comments
 (0)