Skip to content

Commit d997cf0

Browse files
committed
Fix checkpointing for declarative streams
1 parent ceebfda commit d997cf0

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

airbyte_cdk/sources/streams/core.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -223,17 +223,17 @@ def read( # type: ignore # ignoring typing for ConnectorStateManager because o
223223
record_counter += 1
224224

225225
checkpoint_interval = self.state_checkpoint_interval
226-
checkpoint = checkpoint_reader.get_checkpoint()
227226
if (
228227
should_checkpoint
229228
and checkpoint_interval
230229
and record_counter % checkpoint_interval == 0
231-
and checkpoint is not None
232230
):
233-
airbyte_state_message = self._checkpoint_state(
234-
checkpoint, state_manager=state_manager
235-
)
236-
yield airbyte_state_message
231+
checkpoint = checkpoint_reader.get_checkpoint()
232+
if checkpoint:
233+
airbyte_state_message = self._checkpoint_state(
234+
checkpoint, state_manager=state_manager
235+
)
236+
yield airbyte_state_message
237237

238238
if internal_config.is_limit_reached(record_counter):
239239
break

0 commit comments

Comments
 (0)