Skip to content

Commit 5c67a42

Browse files
author
Oleksandr Bazarnov
committed
updated after the review
1 parent 483db58 commit 5c67a42

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

airbyte_cdk/sources/declarative/async_job/job_orchestrator.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -370,7 +370,7 @@ def _process_running_partitions_and_yield_completed_ones(
370370
self._reallocate_partition(current_running_partitions, partition)
371371

372372
# We only remove completed / timeout jobs jobs as we want failed jobs to be re-allocated in priority
373-
self._remove_completed_or_timed_out_jobs(partition)
373+
self._remove_completed_jobs(partition)
374374

375375
# update the referenced list with running partitions
376376
self._running_partitions = current_running_partitions
@@ -395,7 +395,7 @@ def _abort_job(self, job: AsyncJob, free_job_allocation: bool = True) -> None:
395395
except Exception as exception:
396396
LOGGER.warning(f"Could not free budget for job {job.api_job_id()}: {exception}")
397397

398-
def _remove_completed_or_timed_out_jobs(self, partition: AsyncPartition) -> None:
398+
def _remove_completed_jobs(self, partition: AsyncPartition) -> None:
399399
"""
400400
Remove completed or timed out jobs from the partition.
401401

0 commit comments

Comments
 (0)