From c26d355de8576c931ef15bd7d4c6f6352d4d5862 Mon Sep 17 00:00:00 2001 From: Maxime Carbonneau-Leclerc <3360483+maxi297@users.noreply.github.com> Date: Tue, 30 Jan 2024 10:03:00 -0500 Subject: [PATCH] Fix Cursor interface change + FileBased change at the same time (#34653) --- .../sources/file_based/stream/concurrent/adapters.py | 1 + .../airbyte_cdk/sources/file_based/stream/concurrent/cursor.py | 3 +++ 2 files changed, 4 insertions(+) diff --git a/airbyte-cdk/python/airbyte_cdk/sources/file_based/stream/concurrent/adapters.py b/airbyte-cdk/python/airbyte_cdk/sources/file_based/stream/concurrent/adapters.py index 731b046217050..404b7b4ab09ce 100644 --- a/airbyte-cdk/python/airbyte_cdk/sources/file_based/stream/concurrent/adapters.py +++ b/airbyte-cdk/python/airbyte_cdk/sources/file_based/stream/concurrent/adapters.py @@ -78,6 +78,7 @@ def create_from_stream( cursor_field=cursor_field, logger=logger, namespace=stream.namespace, + cursor=cursor, ), stream, cursor, diff --git a/airbyte-cdk/python/airbyte_cdk/sources/file_based/stream/concurrent/cursor.py b/airbyte-cdk/python/airbyte_cdk/sources/file_based/stream/concurrent/cursor.py index a0fd47044f3d7..f91e3a16f3b94 100644 --- a/airbyte-cdk/python/airbyte_cdk/sources/file_based/stream/concurrent/cursor.py +++ b/airbyte-cdk/python/airbyte_cdk/sources/file_based/stream/concurrent/cursor.py @@ -83,5 +83,8 @@ def observe(self, record: Record) -> None: def close_partition(self, partition: Partition) -> None: return None + def ensure_at_least_one_state_emitted(self) -> None: + return None + def set_pending_partitions(self, partitions: Iterable[Partition]) -> None: return None