From 466806529f6d3ae9680925058ff2197642e31121 Mon Sep 17 00:00:00 2001 From: Artem Inzhyyants Date: Wed, 13 Nov 2024 17:23:31 +0100 Subject: [PATCH] Airbyte CDK: update lint Signed-off-by: Artem Inzhyyants --- .../migrations/legacy_to_per_partition_state_migration.py | 4 +--- .../sources/declarative/parsers/model_to_component_factory.py | 2 +- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/airbyte_cdk/sources/declarative/migrations/legacy_to_per_partition_state_migration.py b/airbyte_cdk/sources/declarative/migrations/legacy_to_per_partition_state_migration.py index a615bca15..8000b1876 100644 --- a/airbyte_cdk/sources/declarative/migrations/legacy_to_per_partition_state_migration.py +++ b/airbyte_cdk/sources/declarative/migrations/legacy_to_per_partition_state_migration.py @@ -51,9 +51,7 @@ def __init__( self._cursor.cursor_field, parameters=self._parameters ).eval(self._config) - def _get_partition_field( - self, partition_router: SubstreamPartitionRouter - ) -> str: + def _get_partition_field(self, partition_router: SubstreamPartitionRouter) -> str: parent_stream_config = partition_router.parent_stream_configs[0] # Retrieve the partition field with a condition, as properties are returned as a dictionary for custom components. diff --git a/airbyte_cdk/sources/declarative/parsers/model_to_component_factory.py b/airbyte_cdk/sources/declarative/parsers/model_to_component_factory.py index 9674dc0a5..d2dd9d9dc 100644 --- a/airbyte_cdk/sources/declarative/parsers/model_to_component_factory.py +++ b/airbyte_cdk/sources/declarative/parsers/model_to_component_factory.py @@ -634,7 +634,7 @@ def create_legacy_to_per_partition_state_migration( ) return LegacyToPerPartitionStateMigration( - partition_router, # type: ignore # was already checked above + partition_router, # type: ignore # was already checked above declarative_stream.incremental_sync, # type: ignore # was already checked. Migration can be applied only to incremental streams. config, declarative_stream.parameters, # type: ignore # different type is expected here Mapping[str, Any], got Dict[str, Any]