Skip to content

Commit f530364

Browse files
committed
chore: resolve comments
1 parent 8a96460 commit f530364

File tree

2 files changed

+5
-8
lines changed

2 files changed

+5
-8
lines changed

airbyte_cdk/sources/declarative/manifest_declarative_source.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -143,10 +143,6 @@ def streams(self, config: Mapping[str, Any]) -> List[Stream]:
143143
if api_budget_model:
144144
self._constructor.set_api_budget(api_budget_model, config)
145145

146-
max_concurrent_async_jobs_model = self._source_config.get("max_concurrent_async_jobs")
147-
if max_concurrent_async_jobs_model:
148-
self._constructor.set_max_concurrent_async_jobs(max_concurrent_async_jobs_model, config)
149-
150146
source_streams = [
151147
self._constructor.create_component(
152148
DeclarativeStreamModel,

airbyte_cdk/sources/declarative/parsers/model_to_component_factory.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -541,8 +541,10 @@ def __init__(
541541
)
542542
self._connector_state_manager = connector_state_manager or ConnectorStateManager()
543543
self._api_budget: Optional[Union[APIBudget, HttpAPIBudget]] = None
544-
self._job_tracker: Optional[JobTracker] = self._create_async_job_tracker(
545-
source_config=source_config
544+
self._job_tracker: Optional[JobTracker] = (
545+
self._create_async_job_tracker(source_config=source_config)
546+
if source_config
547+
else None
546548
)
547549

548550
def _init_mappings(self) -> None:
@@ -2928,8 +2930,7 @@ def _get_download_retriever() -> SimpleRetrieverTestReadDecorator | SimpleRetrie
29282930
download_target_extractor=download_target_extractor,
29292931
)
29302932

2931-
if not self._job_tracker:
2932-
self._job_tracker = JobTracker(1)
2933+
self._job_tracker = JobTracker(1) if not self._job_tracker else self._job_tracker
29332934

29342935
async_job_partition_router = AsyncJobPartitionRouter(
29352936
job_orchestrator_factory=lambda stream_slices: AsyncJobOrchestrator(

0 commit comments

Comments
 (0)